Searched defs:while_cond (Results 1 – 8 of 8) sorted by relevance
252 auto* while_cond = while_op->while_condition(); in GetLoopInductionVarTupleIdx() local406 auto* while_cond = while_op->while_condition(); in PatternMatchLoopTripCount() local511 auto* while_cond = while_op->while_condition(); in ComputeWhileLoopTripCount() local580 auto* while_cond = while_op->while_condition(); in ComputeWhileLoopTripCountUpperBound() local
53 HloComputation* while_cond = while_op->while_condition(); in RemoveDeadTupleIndices() local225 HloComputation* while_cond = while_op->while_condition(); in TryRemoveDeadWhileParams() local337 HloComputation* while_cond = while_op->while_condition(); in TryRemoveRepeatedWhileTupleIndicesHelper() local399 HloComputation* while_cond = while_op->while_condition(); in TryRemoveRepeatedWhileTupleIndices() local496 auto* while_cond = while_op->while_condition(); in TryRemoveConstantParams() local859 auto* while_cond = while_op->while_condition(); in TryFlattenNestedTuples() local986 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
1816 auto* while_cond = xla_while->while_condition(); in HandleWhile() local
139 while_cond = while_stmt.cond variable
413 def while_cond(iterate_index, *loop_vars): function477 def while_cond(iterate_index, *loop_vars): function532 def while_cond(iterate, *loop_vars): function614 def while_cond(has_next, *loop_vars): function