Home
last modified time | relevance | path

Searched refs:output_shape_with_layout (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/
H A Dshape_util.cc1537 Shape output_shape_with_layout = output_shape; in AlignLayouts() local
1538 *output_shape_with_layout.mutable_layout() = Layout{layout}; in AlignLayouts()
1539 return output_shape_with_layout; in AlignLayouts()
1578 Shape output_shape_with_layout = MakeShapeWithLayout( in AlignLayouts() local
1580 CHECK(ReshapeIsBitcast(input_shape, output_shape_with_layout)) in AlignLayouts()
1584 << ShapeUtil::HumanStringWithLayout(output_shape_with_layout); in AlignLayouts()
1585 return output_shape_with_layout; in AlignLayouts()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dlayout_assignment_test.cc409 Shape output_shape_with_layout(cshape); in TEST_F() local
411 *output_shape_with_layout.mutable_layout() = in TEST_F()
418 ShapeLayout(output_shape_with_layout); in TEST_F()
H A Dlayout_assignment.cc1290 Shape output_shape_with_layout = ShapeUtil::MakeShapeWithLayout( in ChooseOperandLayoutFromOutputLayout() local
1297 ShapeUtil::AlignLayouts(output_shape_with_layout, operand_shape); in ChooseOperandLayoutFromOutputLayout()