Searched defs:while_init (Results 1 – 7 of 7) sorted by relevance
52 HloInstruction* while_init = while_op->mutable_operand(0); in RemoveDeadTupleIndices() local224 HloInstruction* while_init = while_op->mutable_operand(0); in TryRemoveDeadWhileParams() local339 HloInstruction* while_init = while_op->mutable_operand(0); in TryRemoveRepeatedWhileTupleIndicesHelper() local398 HloInstruction* while_init = while_op->mutable_operand(0); in TryRemoveRepeatedWhileTupleIndices() local494 auto* while_init = while_op->mutable_operand(0); in TryRemoveConstantParams() local719 auto while_init = while_op->operand(0); in TryPropagateConstant() local857 auto* while_init = while_op->mutable_operand(0); in TryFlattenNestedTuples() local984 auto* while_init = while_op->mutable_operand(0); in TryMergeInductionVariables() local
298 auto* while_init = while_op->operand(0); in GetLoopInductionVarTupleIdx() local489 auto* while_init = while_op->mutable_operand(0); in ComputeWhileLoopTripCount() local
347 HloInstruction* while_init{nullptr}; member
300 HloInstruction* while_init = xla_while->mutable_operand(0); in AddCopiesForWhile() local
168 HloInstruction* while_init = in TEST_F() local
1956 auto while_init = builder.AddInstruction(HloInstruction::CreateTuple( in TEST_F() local
2494 HloInstruction* HloInstruction::while_init() const { in while_init() function in xla::HloInstruction