Searched refs:_TPUInferenceContext (Results 1 – 1 of 1) sorted by relevance
1441 if isinstance(context, _TPUInferenceContext):1448 class _TPUInferenceContext(control_flow_ops.XLAControlFlowContext): class1456 super(_TPUInferenceContext, self).__init__()1554 context = _TPUInferenceContext(