Searched defs:while_cond (Results 1 – 9 of 9) sorted by relevance
63 HloComputation* while_cond = while_op->while_condition(); in RemoveDeadTupleIndices() local238 HloComputation* while_cond = while_op->while_condition(); in TryRemoveDeadWhileParams() local474 HloComputation* while_cond = while_op->while_condition(); in TryRemoveRepeatedWhileTupleIndicesHelper() local536 HloComputation* while_cond = while_op->while_condition(); in TryRemoveRepeatedWhileTupleIndices() local633 auto* while_cond = while_op->while_condition(); in TryRemoveConstantParams() local1002 auto* while_cond = while_op->while_condition(); in TryFlattenNestedTuples() local1133 auto* while_cond = while_op->while_condition(); in TryMergeInductionVariables() local
263 auto* while_cond = while_op->while_condition(); in GetLoopInductionVarTupleIdx() local417 auto* while_cond = while_op->while_condition(); in PatternMatchLoopTripCount() local522 auto* while_cond = while_op->while_condition(); in ComputeWhileLoopTripCount() local592 auto* while_cond = while_op->while_condition(); in ComputeWhileLoopTripCountUpperBound() local
155 HloComputation* while_cond = in TEST_F() local191 HloComputation* while_cond = in TEST_F() local223 HloComputation* while_cond = in TEST_F() local
69 HloComputation* while_cond = while_instr->while_condition(); in TrySinkingConstantsIntoWhileLoop() local
234 auto* while_cond = while_op->while_condition(); in TEST_F() local
2397 auto* while_cond = xla_while->while_condition(); in HandleWhile() local
466 HloComputation* while_cond = while_op->while_condition(); in PatternMatchParseWhileLoop() local
139 while_cond = while_stmt.cond variable
436 func::FuncOp while_cond = while_op.cond_function(); in ConvertControlToDataOutputs() local