Searched refs:operand_layout (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | layout_assignment.cc | 832 const ShapeLayout& operand_layout, HloInstruction* instruction, in CopyOperandIfLayoutsDiffer() argument
835 TF_RET_CHECK(operand_layout.LayoutIsSet()); in CopyOperandIfLayoutsDiffer()
838 if (ShapeUtil::Equal(operand_layout.shape(), operand->shape())) { in CopyOperandIfLayoutsDiffer()
845 << operand_layout.ToString() << " in " << instruction->ToString(); in CopyOperandIfLayoutsDiffer()
848 CreateCopyWithNewLayout(operand_layout.shape(), operand)); in CopyOperandIfLayoutsDiffer()
1028 auto operand_layout = aligned_operand_shape.value().layout(); in ChooseOperandLayoutFromOutputLayout() local
1030 LayoutUtil::ValidateLayoutForShape(operand_layout, operand_shape)); in ChooseOperandLayoutFromOutputLayout()
1031 return absl::make_unique<Layout>(operand_layout); in ChooseOperandLayoutFromOutputLayout()
1044 Layout operand_layout = LayoutUtil::MakeLayout(new_minor_to_major); in ChooseOperandLayoutFromOutputLayout() local
1046 LayoutUtil::ValidateLayoutForShape(operand_layout, operand->shape())); in ChooseOperandLayoutFromOutputLayout()
[all …]
|
| D | layout_assignment.h | 370 const Layout& operand_layout, const HloInstruction* user,
469 Status CopyOperandIfLayoutsDiffer(const ShapeLayout& operand_layout,
|
| D | hlo_verifier.cc | 1447 const Layout& operand_layout = operand_shape.layout(); in Postprocess() local
1448 TF_RET_CHECK(LayoutUtil::Equal(result_layout, operand_layout)) in Postprocess()
|