Home
last modified time | relevance | path

Searched defs:operand_layout (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dlayout_assignment.cc1050 const ShapeLayout& operand_layout, HloInstruction* instruction, in CopyOperandIfLayoutsDiffer()
1299 auto operand_layout = aligned_operand_shape.value().layout(); in ChooseOperandLayoutFromOutputLayout() local
1315 Layout operand_layout = LayoutUtil::MakeLayout(new_minor_to_major); in ChooseOperandLayoutFromOutputLayout() local
1324 static Layout GetReduceLayoutFromOperand(const Layout& operand_layout, in GetReduceLayoutFromOperand()
1334 const Layout& operand_layout, const HloInstruction* user, in ChooseOutputLayoutFromOperandLayout()
1765 std::unique_ptr<Layout> operand_layout = in PropagateBufferConstraintToOperands() local
2014 const ShapeLayout* operand_layout = in AssignLayouts() local
H A Dhlo_verifier.cc2521 const Layout& operand_layout = operand_shape.layout(); in Postprocess() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dreduction_layout_normalizer.cc70 const Layout &operand_layout = operand_shape.layout(); in HandleReduce() local
/aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/expansions/
H A Dmeta_spmd_expander.cc345 Status VerifyTileOperandLayout(const Layout& operand_layout, in VerifyTileOperandLayout()