Searched defs:while_init (Results 1 – 7 of 7) sorted by relevance
56 HloInstruction* while_init = while_op->mutable_operand(0); in TryRemoveDeadWhileParams() local313 auto* while_init = while_op->mutable_operand(0); in TryRemoveConstantParams() local507 auto while_init = while_op->operand(0); in TryPropagateConstant() local645 auto* while_init = while_op->mutable_operand(0); in TryFlattenNestedTuples() local772 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
275 HloInstruction* while_init = in TEST_F() local
277 HloInstruction* while_init = xla_while->mutable_operand(0); in AddCopiesForWhile() local
167 HloInstruction* while_init = in TEST_F() local
1753 auto while_init = builder.AddInstruction(HloInstruction::CreateTuple( in TEST_F() local
2052 HloInstruction* HloInstruction::while_init() const { in while_init() function in xla::HloInstruction