Searched refs:XLAControlFlowContext (Results 1 – 9 of 9) sorted by relevance
138 xla_context = control_flow_ops.XLAControlFlowContext()172 xla_context = control_flow_ops.XLAControlFlowContext()233 xla_context = control_flow_ops.XLAControlFlowContext()
52 context, control_flow_ops.XLAControlFlowContext):
124 class TPUReplicateContext(control_flow_ops.XLAControlFlowContext):1448 class _TPUInferenceContext(control_flow_ops.XLAControlFlowContext):
132 class CatchInvalidHostcallFunctions(control_flow_ops.XLAControlFlowContext):
101 class XLACompileContext(control_flow_ops.XLAControlFlowContext):
699 xla_context = control_flow_ops.XLAControlFlowContext()736 xla_context = control_flow_ops.XLAControlFlowContext()
1388 xla_context = control_flow_ops.XLAControlFlowContext()1425 xla_context = control_flow_ops.XLAControlFlowContext()1467 xla_context = control_flow_ops.XLAControlFlowContext()1474 xla_context = control_flow_ops.XLAControlFlowContext()1525 xla_context = control_flow_ops.XLAControlFlowContext()
4016 class XLAControlFlowContext(ControlFlowContext): class4020 super(XLAControlFlowContext, self).__init__()4026 super(XLAControlFlowContext, self).to_control_flow_context_def(
760 tpu_context, control_flow_ops.XLAControlFlowContext):