Searched refs:AlignLayouts (Results 1 – 4 of 4) sorted by relevance
742 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()
1260 /* static */ absl::optional<Shape> ShapeUtil::AlignLayouts( in AlignLayouts() function in xla::ShapeUtil1269 AlignLayouts(DropDegenerateDimensions(input_shape), in AlignLayouts()
603 static absl::optional<Shape> AlignLayouts(const Shape& input_shape,
1026 ShapeUtil::AlignLayouts(output_shape_with_layout, operand_shape); in ChooseOperandLayoutFromOutputLayout()1087 ShapeUtil::AlignLayouts(operand_shape_with_layout, output_shape); in ChooseOutputLayoutFromOperandLayout()