Searched defs:while_init (Results 1 – 7 of 7) sorted by relevance
67 HloInstruction* while_init = while_op->mutable_operand(0); in RemoveDeadTupleIndices() local241 HloInstruction* while_init = while_op->mutable_operand(0); in TryRemoveDeadWhileParams() local372 HloInstruction* while_init = while_op->mutable_operand(0); in TryRemoveRepeatedWhileTupleIndicesHelper() local431 HloInstruction* while_init = while_op->mutable_operand(0); in TryRemoveRepeatedWhileTupleIndices() local527 auto* while_init = while_op->mutable_operand(0); in TryRemoveConstantParams() local757 auto while_init = while_op->operand(0); in TryPropagateConstant() local895 auto* while_init = while_op->mutable_operand(0); in TryFlattenNestedTuples() local1025 auto* while_init = while_op->mutable_operand(0); in TryMergeInductionVariables() local
302 auto* while_init = while_op->operand(0); in GetLoopInductionVarTupleIdx() local493 auto* while_init = while_op->mutable_operand(0); in ComputeWhileLoopTripCount() local
347 HloInstruction* while_init{nullptr}; member
168 HloInstruction* while_init = in TEST_F() local
305 HloInstruction* while_init = xla_while->mutable_operand(0); in AddCopiesForWhile() local
1956 auto while_init = builder.AddInstruction(HloInstruction::CreateTuple( in TEST_F() local
2694 HloInstruction* HloInstruction::while_init() const { in while_init() function in xla::HloInstruction