Searched refs:producing_instruction (Results 1 – 1 of 1) sorted by relevance
2737 HloInstruction* producing_instruction = AddGetTupleElements(); in Process() local2738 HloComputation* computation = producing_instruction->parent(); in Process()2741 HloInstruction* replacement_instruction = producing_instruction; in Process()2745 ReplaceTupleWith(producing_instruction, in Process()2748 } else if (operand_shape != producing_instruction->shape()) { in Process()2750 << ", new shape = " << producing_instruction->shape().ToString() in Process()2753 HloInstruction::CreateBitcast(operand_shape, producing_instruction)); in Process()2821 HloInstruction* producing_instruction = defining_position().instruction; in AddGetTupleElements() local2822 CHECK_NE(producing_instruction, nullptr); in AddGetTupleElements()2828 HloComputation* computation = producing_instruction->parent(); in AddGetTupleElements()[all …]