Searched refs:instruction_to_append (Results 1 – 4 of 4) sorted by relevance
1609 HloInstruction* instruction_to_append, bool add_output) { in AppendInstructionIntoCalledComputation() argument1613 CHECK(IsUserOf(instruction_to_append)); in AppendInstructionIntoCalledComputation()1615 return CloneAndAppendInstructionIntoCalledComputation(instruction_to_append, in AppendInstructionIntoCalledComputation()1621 HloInstruction* instruction_to_append, bool add_output) { in CloneAndAppendInstructionIntoCalledComputation() argument1622 CHECK(instruction_to_append->IsFusible()) in CloneAndAppendInstructionIntoCalledComputation()1623 << instruction_to_append->ToString(); in CloneAndAppendInstructionIntoCalledComputation()1625 << instruction_to_append->ToString(); in CloneAndAppendInstructionIntoCalledComputation()1633 builder.AddInstruction(instruction_to_append->Clone(/*suffix=*/"")); in CloneAndAppendInstructionIntoCalledComputation()1643 absl::c_linear_search(operands(), instruction_to_append); in CloneAndAppendInstructionIntoCalledComputation()1645 if (instruction_to_append->opcode() == HloOpcode::kTuple) { in CloneAndAppendInstructionIntoCalledComputation()[all …]
1170 HloInstruction* instruction_to_append, bool add_output = false);1176 HloInstruction* instruction_to_append, bool add_output = false);
1928 HloInstruction* instruction_to_append, bool add_output = false);
4549 HloInstruction* instruction_to_append, bool add_output) { in AppendInstructionIntoCalledComputation() argument4551 ->AppendInstructionIntoCalledComputation(instruction_to_append, in AppendInstructionIntoCalledComputation()