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()
|