Searched defs:while_init (Results 1 – 6 of 6) sorted by relevance
56 HloInstruction* while_init = while_op->mutable_operand(0); in TryRemoveDeadWhileParams() local312 auto* while_init = while_op->mutable_operand(0); in TryRemoveConstantParams() local508 auto while_init = while_op->operand(0); in TryPropagateConstant() local646 auto* while_init = while_op->mutable_operand(0); in TryFlattenNestedTuples() local773 auto* while_init = while_op->mutable_operand(0); in TryMergeInductionVariables() local
156 auto* while_init = while_op->operand(0); in GetLoopInductionVarTupleIdx() local347 auto* while_init = while_op->mutable_operand(0); in ComputeWhileLoopTripCount() local
277 HloInstruction* while_init = xla_while->mutable_operand(0); in AddCopiesForWhile() local
167 HloInstruction* while_init = in TEST_F() local
1764 auto while_init = builder.AddInstruction(HloInstruction::CreateTuple( in TEST_F() local
2193 HloInstruction* HloInstruction::while_init() const { in while_init() function in xla::HloInstruction