Searched refs:copy_leaf (Results 1 – 2 of 2) sorted by relevance
799 HloComputation* computation)>& copy_leaf) { in DeepCopyHelper() argument811 DeepCopyHelper(gte, index, copy_leaf)); in DeepCopyHelper()825 return copy_leaf(instruction, *index, this); in DeepCopyHelper()846 auto copy_leaf = [indices_to_copy, copies_added]( in DeepCopyInstruction() local861 return DeepCopyHelper(instruction, &index, copy_leaf); in DeepCopyInstruction()868 HloComputation* computation)>& copy_leaf) { in DeepCopyInstructionWithCustomCopier() argument875 return DeepCopyHelper(instruction, &index, copy_leaf); in DeepCopyInstructionWithCustomCopier()
379 HloComputation* computation)>& copy_leaf);691 HloComputation* computation)>& copy_leaf);