Searched refs:HasDegenerateDimensions (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | shape_util_test.cc | 680 TEST(ShapeUtilTest, HasDegenerateDimensions) { in TEST() argument
682 ShapeUtil::HasDegenerateDimensions(ShapeUtil::MakeShape(F32, {3, 1, 2}))); in TEST()
684 ShapeUtil::HasDegenerateDimensions(ShapeUtil::MakeShape(F32, {3, 1, 1}))); in TEST()
686 ShapeUtil::HasDegenerateDimensions(ShapeUtil::MakeShape(F32, {3, 3, 5}))); in TEST()
688 ShapeUtil::HasDegenerateDimensions(ShapeUtil::MakeShape(F32, {3, 0, 5}))); in TEST()
|
| D | shape_util.cc | 844 /* static */ bool ShapeUtil::HasDegenerateDimensions(const Shape& shape) { in HasDegenerateDimensions() function in xla::ShapeUtil
1266 if (HasDegenerateDimensions(input_shape) || in AlignLayouts()
1267 HasDegenerateDimensions(output_shape)) { in AlignLayouts()
|
| D | shape_util.h | 532 static bool HasDegenerateDimensions(const Shape& shape);
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | indexed_array_analysis.cc | 469 if (!ShapeUtil::HasDegenerateDimensions(shape)) { in ReshapeToRemoveDegenerateDims()
528 CHECK(!ShapeUtil::HasDegenerateDimensions(operand->shape())); in ReshapeToAddDegenerateDims()
638 CHECK(!ShapeUtil::HasDegenerateDimensions(shape)); in FoldReshapeOfGatherNoDegenerateDims()
639 CHECK(!ShapeUtil::HasDegenerateDimensions(scalar_indexed->shape())); in FoldReshapeOfGatherNoDegenerateDims()
|