Searched defs:while_loop (Results 1 – 12 of 12) sorted by relevance
77 def while_loop(x): function
1516 def while_loop(): function
30 def while_loop(condition: Callable[..., Any], function
3922 const auto while_loop = root->operand(0); in TEST_F() local3974 const auto while_loop = root->operand(0); in TEST_F() local4359 const auto while_loop = root->operand(0); in TEST_F() local4437 const auto while_loop = root->operand(1)->operand(0); in TEST_F() local4512 const auto while_loop = root->operand(0)->operand(0); in TEST_F() local4603 const auto while_loop = root->operand(0); in TEST_F() local4682 const auto while_loop = root->operand(1)->operand(0); in TEST_F() local4759 const auto while_loop = root->operand(0)->operand(0); in TEST_F() local4824 const auto while_loop = root->operand(0); in TEST_F() local4963 const auto while_loop = in TEST_F() local[all …]
326 auto while_loop = b->AddInstruction(HloInstruction::CreateWhile( in GetFinalFftUsingCollectivePermute() local
578 HloInstruction* while_loop; member
707 std::vector<ReplicaGroup> GetLoopReplicaGroups(HloInstruction* while_loop) { in GetLoopReplicaGroups()1771 auto while_loop = b->AddInstruction(HloInstruction::CreateWhile( in PartitionBaseCase() local
71 def while_loop(cond, function
2528 def while_loop(cond, function
561 while_loop = gen_xla_ops.xla_while variable
171 HloInstruction* while_loop = in TEST_F() local
917 auto while_loop = [this, &body, build_condition](int count) { in XLA_TEST_F() local