Searched defs:while_loop (Results 1 – 15 of 15) sorted by relevance
34 void while_loop() { in while_loop() function
81 def while_loop(x): function
1501 def while_loop(): function
136 void while_loop() { in while_loop() function
34 def while_loop(condition: Callable[..., Any], function
80 While while_loop = 3; field
328 auto while_loop = b->AddInstruction(HloInstruction::CreateWhile( in GetFinalFftUsingCollectivePermute() local
505 HloInstruction* while_loop; member
3872 auto while_loop = root->operand(0); in TEST_F() local3966 auto while_loop = root->operand(0); in TEST_F() local4033 auto while_loop = root->operand(0)->operand(0); in TEST_F() local4097 auto while_loop = root->operand(0); in TEST_F() local
1419 auto while_loop = b->AddInstruction(HloInstruction::CreateWhile( in PartitionBaseCase() local
60 def while_loop(cond, function
2546 def while_loop(cond, function
478 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