Searched defs:while_cond (Results 1 – 8 of 8) sorted by relevance
256 auto* while_cond = while_op->while_condition(); in GetLoopInductionVarTupleIdx() local410 auto* while_cond = while_op->while_condition(); in PatternMatchLoopTripCount() local515 auto* while_cond = while_op->while_condition(); in ComputeWhileLoopTripCount() local584 auto* while_cond = while_op->while_condition(); in ComputeWhileLoopTripCountUpperBound() local
68 HloComputation* while_cond = while_op->while_condition(); in RemoveDeadTupleIndices() local242 HloComputation* while_cond = while_op->while_condition(); in TryRemoveDeadWhileParams() local370 HloComputation* while_cond = while_op->while_condition(); in TryRemoveRepeatedWhileTupleIndicesHelper() local432 HloComputation* while_cond = while_op->while_condition(); in TryRemoveRepeatedWhileTupleIndices() local529 auto* while_cond = while_op->while_condition(); in TryRemoveConstantParams() local897 auto* while_cond = while_op->while_condition(); in TryFlattenNestedTuples() local1027 auto* while_cond = while_op->while_condition(); in TryMergeInductionVariables() local
51 HloComputation* while_cond = while_instr->while_condition(); in TrySinkingConstantsIntoWhileLoop() local
155 HloComputation* while_cond = in TEST_F() local191 HloComputation* while_cond = in TEST_F() local223 HloComputation* while_cond = in TEST_F() local
234 auto* while_cond = while_op->while_condition(); in TEST_F() local
2168 auto* while_cond = xla_while->while_condition(); in HandleWhile() local
139 while_cond = while_stmt.cond variable
412 def while_cond(iterate_index, *loop_vars): function476 def while_cond(iterate_index, *loop_vars): function531 def while_cond(iterate, *loop_vars): function613 def while_cond(has_next, *loop_vars): function