Home
last modified time | relevance | path

Searched refs:dim_to_delete (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/
Dshape.cc97 void Shape::DeleteDimension(int64 dim_to_delete) { in DeleteDimension() argument
99 CHECK_GE(dim_to_delete, 0); in DeleteDimension()
100 CHECK_LT(dim_to_delete, dimensions_.size()); in DeleteDimension()
101 dimensions_.erase(dimensions_.begin() + dim_to_delete); in DeleteDimension()
102 dynamic_dimensions_.erase(dynamic_dimensions_.begin() + dim_to_delete); in DeleteDimension()
106 if (layout_.minor_to_major(i) == dim_to_delete) { in DeleteDimension()
111 if (layout_.minor_to_major(i) > dim_to_delete) { in DeleteDimension()
Dshape.h83 void DeleteDimension(int64 dim_to_delete);
Dshape_util.cc1410 /* static */ Shape ShapeUtil::DeleteDimension(int64 dim_to_delete, in DeleteDimension() argument
1413 shape.DeleteDimension(dim_to_delete); in DeleteDimension()
Dshape_util.h609 static Shape DeleteDimension(int64 dim_to_delete, Shape shape);