Searched defs:while_loop (Results 1 – 12 of 12) sorted by relevance
81 def while_loop(x): function
1530 def while_loop(): function
34 def while_loop(condition: Callable[..., Any], function
328 auto while_loop = b->AddInstruction(HloInstruction::CreateWhile( in GetFinalFftUsingCollectivePermute() local
526 HloInstruction* while_loop; member
4126 auto while_loop = root->operand(0); in TEST_F() local4220 auto while_loop = root->operand(0); in TEST_F() local4289 auto while_loop = root->operand(0); in TEST_F() local4356 auto while_loop = root->operand(0)->operand(0); in TEST_F() local4420 auto while_loop = root->operand(0); in TEST_F() local
461 std::vector<ReplicaGroup> GetLoopReplicaGroups(HloInstruction* while_loop) { in GetLoopReplicaGroups()1508 auto while_loop = b->AddInstruction(HloInstruction::CreateWhile( in PartitionBaseCase() local
74 def while_loop(cond, function
2563 def while_loop(cond, function
546 while_loop = gen_xla_ops.xla_while variable
172 HloInstruction* while_loop = in TEST_F() local
932 auto while_loop = [this, &body, build_condition](int count) { in XLA_TEST_F() local