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.cc742 auto aligned_shape = ShapeUtil::AlignLayouts( in TEST()
749 aligned_shape = ShapeUtil::AlignLayouts( in TEST()
761 auto aligned_shape = ShapeUtil::AlignLayouts( in TEST()
770 auto aligned_shape = ShapeUtil::AlignLayouts( in TEST()
783 auto aligned_shape = ShapeUtil::AlignLayouts( in TEST()
794 auto aligned_shape = ShapeUtil::AlignLayouts( in TEST()
Dshape_util.cc1260 /* static */ absl::optional<Shape> ShapeUtil::AlignLayouts( in AlignLayouts() function in xla::ShapeUtil
1269 AlignLayouts(DropDegenerateDimensions(input_shape), in AlignLayouts()
Dshape_util.h603 static absl::optional<Shape> AlignLayouts(const Shape& input_shape,
/external/tensorflow/tensorflow/compiler/xla/service/
Dlayout_assignment.cc1026 ShapeUtil::AlignLayouts(output_shape_with_layout, operand_shape); in ChooseOperandLayoutFromOutputLayout()
1087 ShapeUtil::AlignLayouts(operand_shape_with_layout, output_shape); in ChooseOutputLayoutFromOperandLayout()