Searched refs:output_shape_with_layout (Results 1 – 3 of 3) sorted by relevance
1296 Shape output_shape_with_layout = output_shape; in AlignLayouts() local1297 *output_shape_with_layout.mutable_layout()->mutable_minor_to_major() = in AlignLayouts()1299 return output_shape_with_layout; in AlignLayouts()1399 Shape output_shape_with_layout = MakeShapeWithLayout( in AlignLayouts() local1402 CHECK(ReshapeIsBitcast(input_shape, output_shape_with_layout)) in AlignLayouts()1406 << ShapeUtil::HumanStringWithLayout(output_shape_with_layout); in AlignLayouts()1407 return output_shape_with_layout; in AlignLayouts()
445 Shape output_shape_with_layout(cshape); in TEST_F() local447 *output_shape_with_layout.mutable_layout() = in TEST_F()454 ShapeLayout(output_shape_with_layout); in TEST_F()
1019 Shape output_shape_with_layout = ShapeUtil::MakeShapeWithLayout( in ChooseOperandLayoutFromOutputLayout() local1026 ShapeUtil::AlignLayouts(output_shape_with_layout, operand_shape); in ChooseOperandLayoutFromOutputLayout()