Searched refs:HasDegenerateDimensions (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | reduction_degenerate_dim_remover.cc | 53 if (!ShapeUtil::HasDegenerateDimensions(reduced_op->shape())) { in HandleReduce()
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | shape_util_test.cc | 718 TEST(ShapeUtilTest, HasDegenerateDimensions) { in TEST() argument
720 ShapeUtil::HasDegenerateDimensions(ShapeUtil::MakeShape(F32, {3, 1, 2}))); in TEST()
722 ShapeUtil::HasDegenerateDimensions(ShapeUtil::MakeShape(F32, {3, 1, 1}))); in TEST()
724 ShapeUtil::HasDegenerateDimensions(ShapeUtil::MakeShape(F32, {3, 3, 5}))); in TEST()
726 ShapeUtil::HasDegenerateDimensions(ShapeUtil::MakeShape(F32, {3, 0, 5}))); in TEST()
|
| D | shape_util.cc | 1050 /* static */ bool ShapeUtil::HasDegenerateDimensions(const Shape& shape) { in HasDegenerateDimensions() function in xla::ShapeUtil
1506 if (HasDegenerateDimensions(input_shape) || in AlignLayouts()
1507 HasDegenerateDimensions(output_shape)) { in AlignLayouts()
|
| D | shape_util.h | 493 static bool HasDegenerateDimensions(const Shape& shape);
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | indexed_array_analysis.cc | 475 if (!ShapeUtil::HasDegenerateDimensions(shape)) { in ReshapeToRemoveDegenerateDims()
534 CHECK(!ShapeUtil::HasDegenerateDimensions(operand->shape())); in ReshapeToAddDegenerateDims()
644 CHECK(!ShapeUtil::HasDegenerateDimensions(shape)); in FoldReshapeOfGatherNoDegenerateDims()
645 CHECK(!ShapeUtil::HasDegenerateDimensions(scalar_indexed->shape())); in FoldReshapeOfGatherNoDegenerateDims()
|
| D | algebraic_simplifier.cc | 3559 if (ShapeUtil::HasDegenerateDimensions(operand->shape())) { in HandleBroadcast()
|