Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/
Dshape_util.cc1296 Shape output_shape_with_layout = output_shape; in AlignLayouts() local
1399 Shape output_shape_with_layout = MakeShapeWithLayout( in AlignLayouts() local
/external/tensorflow/tensorflow/compiler/xla/service/
Dlayout_assignment.cc1019 Shape output_shape_with_layout = ShapeUtil::MakeShapeWithLayout( in ChooseOperandLayoutFromOutputLayout() local
Dlayout_assignment_test.cc445 Shape output_shape_with_layout(cshape); in TEST_F() local