Searched refs:dim_to_delete (Results 1 – 4 of 4) sorted by relevance
122 void Shape::DeleteDimension(int64_t dim_to_delete) { in DeleteDimension() argument124 CHECK_GE(dim_to_delete, 0); in DeleteDimension()125 CHECK_LT(dim_to_delete, dimensions_.size()); in DeleteDimension()126 dimensions_.erase(dimensions_.begin() + dim_to_delete); in DeleteDimension()127 dynamic_dimensions_.erase(dynamic_dimensions_.begin() + dim_to_delete); in DeleteDimension()130 if (layout_->minor_to_major(i) == dim_to_delete) { in DeleteDimension()135 if (layout_->minor_to_major(i) > dim_to_delete) { in DeleteDimension()
103 void DeleteDimension(int64_t dim_to_delete);
594 static Shape DeleteDimension(int64_t dim_to_delete, Shape shape);
1588 /* static */ Shape ShapeUtil::DeleteDimension(int64_t dim_to_delete, in DeleteDimension() argument1591 shape.DeleteDimension(dim_to_delete); in DeleteDimension()