Searched refs:XLAControlFlowContext (Results 1 – 3 of 3) sorted by relevance
109 class TPUReplicateContext(control_flow_ops.XLAControlFlowContext):
791 xla_context = control_flow_ops.XLAControlFlowContext()825 xla_context = control_flow_ops.XLAControlFlowContext()865 xla_context = control_flow_ops.XLAControlFlowContext()913 xla_context = control_flow_ops.XLAControlFlowContext()
3702 class XLAControlFlowContext(ControlFlowContext): class3706 super(XLAControlFlowContext, self).__init__()