Home
last modified time | relevance | path

Searched refs:AlignLayouts (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/
Dshape_util_test.cc773 auto aligned_shape = ShapeUtil::AlignLayouts( in TEST()
780 aligned_shape = ShapeUtil::AlignLayouts( in TEST()
792 auto aligned_shape = ShapeUtil::AlignLayouts( in TEST()
801 auto aligned_shape = ShapeUtil::AlignLayouts( in TEST()
814 auto aligned_shape = ShapeUtil::AlignLayouts( in TEST()
825 auto aligned_shape = ShapeUtil::AlignLayouts( in TEST()
Dshape_util.cc1287 /* static */ absl::optional<Shape> ShapeUtil::AlignLayouts( in AlignLayouts() function in xla::ShapeUtil
1296 AlignLayouts(DropDegenerateDimensions(input_shape), in AlignLayouts()
Dshape_util.h630 static absl::optional<Shape> AlignLayouts(const Shape& input_shape,
/external/tensorflow/tensorflow/compiler/xla/service/
Dlayout_assignment.cc1083 ShapeUtil::AlignLayouts(output_shape_with_layout, operand_shape); in ChooseOperandLayoutFromOutputLayout()
1144 ShapeUtil::AlignLayouts(operand_shape_with_layout, output_shape); in ChooseOutputLayoutFromOperandLayout()