Searched refs:SetupCopiedInstruction (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | layout_assignment.h | 464 static void SetupCopiedInstruction(const HloInstruction& instruction,
|
| D | layout_assignment.cc | 824 SetupCopiedInstruction(*instruction, gte, {i}); in CreateCopyWithNewLayout()
834 SetupCopiedInstruction(*instruction, tuple_copy, {}); in CreateCopyWithNewLayout()
844 SetupCopiedInstruction(*instruction, copy, {}); in CreateCopyWithNewLayout()
908 void LayoutAssignment::SetupCopiedInstruction(const HloInstruction& instruction, in SetupCopiedInstruction() function in xla::LayoutAssignment
2376 SetupCopiedInstruction(*operand, copy, {}); in AddCopyForOperand()
|