Searched refs:SetupCopiedInstruction (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | layout_assignment.cc | 796 SetupCopiedInstruction(*instruction, gte, {i}); in CreateCopyWithNewLayout()
806 SetupCopiedInstruction(*instruction, tuple_copy, {}); in CreateCopyWithNewLayout()
816 SetupCopiedInstruction(*instruction, copy, {}); in CreateCopyWithNewLayout()
855 void LayoutAssignment::SetupCopiedInstruction(const HloInstruction& instruction, in SetupCopiedInstruction() function in xla::LayoutAssignment
1850 SetupCopiedInstruction(*operand, copy, {}); in ConstrainChannelLayouts()
1876 SetupCopiedInstruction(*operand, copy, {}); in ConstrainChannelLayouts()
2134 SetupCopiedInstruction(*operand, copy, {}); in AddCopyForOperand()
|
| D | layout_assignment.h | 456 static void SetupCopiedInstruction(const HloInstruction& instruction,
|