Searched refs:copy_leaf (Results 1 – 2 of 2) sorted by relevance
756 HloComputation* computation)>& copy_leaf) { in DeepCopyHelper() argument768 DeepCopyHelper(gte, index, copy_leaf)); in DeepCopyHelper()782 return copy_leaf(instruction, *index, this); in DeepCopyHelper()803 auto copy_leaf = [indices_to_copy, copies_added]( in DeepCopyInstruction() local818 return DeepCopyHelper(instruction, &index, copy_leaf); in DeepCopyInstruction()825 HloComputation* computation)>& copy_leaf) { in DeepCopyInstructionWithCustomCopier() argument832 return DeepCopyHelper(instruction, &index, copy_leaf); in DeepCopyInstructionWithCustomCopier()
307 HloComputation* computation)>& copy_leaf);522 HloComputation* computation)>& copy_leaf);