Home
last modified time | relevance | path

Searched refs:ValidateLayoutInShape (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/
Dlayout_util_test.cc350 LayoutUtil::ValidateLayoutInShape(shape, /*allow_missing_layouts=*/false); in TEST_F()
353 LayoutUtil::ValidateLayoutInShape(shape, /*allow_missing_layouts=*/true); in TEST_F()
361 LayoutUtil::ValidateLayoutInShape(shape, /*allow_missing_layouts=*/false); in TEST_F()
367 LayoutUtil::ValidateLayoutInShape(shape, /*allow_missing_layouts=*/true); in TEST_F()
378 LayoutUtil::ValidateLayoutInShape(shape, /*allow_missing_layouts=*/false); in TEST_F()
383 LayoutUtil::ValidateLayoutInShape(shape, /*allow_missing_layouts=*/true); in TEST_F()
391 LayoutUtil::ValidateLayoutInShape(shape, /*allow_missing_layouts=*/false); in TEST_F()
396 LayoutUtil::ValidateLayoutInShape(shape, /*allow_missing_layouts=*/true); in TEST_F()
414 LayoutUtil::ValidateLayoutInShape(shape, /*allow_missing_layouts=*/false); in TEST_F()
419 LayoutUtil::ValidateLayoutInShape(shape, /*allow_missing_layouts=*/true); in TEST_F()
[all …]
Dlayout_util.h79 static Status ValidateLayoutInShape(const Shape& shape,
Dlayout_util.cc175 /* static */ Status LayoutUtil::ValidateLayoutInShape( in ValidateLayoutInShape() function in xla::LayoutUtil
184 ValidateLayoutInShape(element_shape, allow_missing_layouts)); in ValidateLayoutInShape()
Dshape_util.cc870 return LayoutUtil::ValidateLayoutInShape(shape, in ValidateShapeWithOptionalLayout()
877 return LayoutUtil::ValidateLayoutInShape(shape); in ValidateShape()
/external/tensorflow/tensorflow/compiler/xla/service/
Dlayout_assignment.cc268 TF_RETURN_IF_ERROR(LayoutUtil::ValidateLayoutInShape(shape)); in SetArrayOperandLayout()