Searched refs:copy_leaf (Results 1 – 2 of 2) sorted by relevance
628 HloComputation* computation)>& copy_leaf) { in DeepCopyHelper() argument640 DeepCopyHelper(gte, index, copy_leaf)); in DeepCopyHelper()654 return copy_leaf(instruction, *index, this); in DeepCopyHelper()675 auto copy_leaf = [indices_to_copy, copies_added]( in DeepCopyInstruction() local690 return DeepCopyHelper(instruction, &index, copy_leaf); in DeepCopyInstruction()697 HloComputation* computation)>& copy_leaf) { in DeepCopyInstructionWithCustomCopier() argument704 return DeepCopyHelper(instruction, &index, copy_leaf); in DeepCopyInstructionWithCustomCopier()
266 HloComputation* computation)>& copy_leaf);443 HloComputation* computation)>& copy_leaf);