Searched refs:IsInXLAContext (Results 1 – 5 of 5) sorted by relevance
52 self.assertFalse(control_flow_util.IsInXLAContext(z1.op))53 self.assertTrue(control_flow_util.IsInXLAContext(z2.op))
195 self.assertFalse(control_flow_util.IsInXLAContext(op1.op))196 self.assertTrue(control_flow_util.IsInXLAContext(op2.op))
58 def IsInXLAContext(op): function
340 if not util.IsInXLAContext(value.op):
1754 if not util.IsInXLAContext(op) and op.type in {"Shape", "Size", "Rank"}: