Searched refs:dim_to_delete (Results 1 – 2 of 2) sorted by relevance
1403 /* static */ Shape ShapeUtil::DeleteDimension(int64 dim_to_delete, in DeleteDimension() argument1405 shape.mutable_dimensions()->erase(shape.dimensions().begin() + dim_to_delete); in DeleteDimension()1410 if (layout->minor_to_major(i) == dim_to_delete) { in DeleteDimension()1415 if (layout->minor_to_major(i) > dim_to_delete) { in DeleteDimension()
546 static Shape DeleteDimension(int64 dim_to_delete, Shape shape);