Searched defs:while_cond (Results 1 – 6 of 6) sorted by relevance
110 auto* while_cond = while_op->while_condition(); in GetLoopInductionVarTupleIdx() local264 auto* while_cond = while_op->while_condition(); in PatternMatchLoopTripCount() local369 auto* while_cond = while_op->while_condition(); in ComputeWhileLoopTripCount() local438 auto* while_cond = while_op->while_condition(); in ComputeWhileLoopTripCountUpperBound() local
103 HloComputation* while_cond, HloComputation* while_body, in MakeRematerializableWhileComputation()235 HloComputation* while_cond = in TEST_F() local271 HloComputation* while_cond = in TEST_F() local303 HloComputation* while_cond = in TEST_F() local
51 HloComputation* while_cond = while_instr->while_condition(); in TrySinkingConstantsIntoWhileLoop() local
57 HloComputation* while_cond = while_op->while_condition(); in TryRemoveDeadWhileParams() local315 auto* while_cond = while_op->while_condition(); in TryRemoveConstantParams() local647 auto* while_cond = while_op->while_condition(); in TryFlattenNestedTuples() local774 auto* while_cond = while_op->while_condition(); in TryMergeInductionVariables() local
1354 auto* while_cond = xla_while->while_condition(); in HandleWhile() local
108 def while_cond(iterate_index, *state): function