Searched refs:in_place_input_source (Results 1 – 1 of 1) sorted by relevance
1735 const HloInstruction* in_place_input_source = nullptr; in GetFusionInstructionInPlaceInputOutputPairs() local1742 CHECK(in_place_input_source == nullptr); in GetFusionInstructionInPlaceInputOutputPairs()1743 in_place_input_source = in GetFusionInstructionInPlaceInputOutputPairs()1749 if (in_place_input_source) { in GetFusionInstructionInPlaceInputOutputPairs()1753 std::tie(in_place_input_source, in_place_input_index) = in GetFusionInstructionInPlaceInputOutputPairs()1754 FollowTupleIndirection(in_place_input_source, in_place_input_index); in GetFusionInstructionInPlaceInputOutputPairs()1756 if (in_place_input_source->opcode() == HloOpcode::kParameter) { in GetFusionInstructionInPlaceInputOutputPairs()1758 HloOperandIndex{in_place_input_source->parameter_number(), in GetFusionInstructionInPlaceInputOutputPairs()