Home
last modified time | relevance | path

Searched refs:SetupCopiedInstruction (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dlayout_assignment.h464 static void SetupCopiedInstruction(const HloInstruction& instruction,
Dlayout_assignment.cc824 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()