Searched refs:DropDegenerateDimensions (Results 1 – 2 of 2) sorted by relevance
849 /* static */ Shape ShapeUtil::DropDegenerateDimensions(const Shape& shape) { in DropDegenerateDimensions() function in xla::ShapeUtil1269 AlignLayouts(DropDegenerateDimensions(input_shape), in AlignLayouts()1270 DropDegenerateDimensions(output_shape)); in AlignLayouts()
535 static Shape DropDegenerateDimensions(const Shape& shape);