Searched refs:DeleteDimension (Results 1 – 3 of 3) sorted by relevance
546 static Shape DeleteDimension(int64 dim_to_delete, Shape shape);
1403 /* static */ Shape ShapeUtil::DeleteDimension(int64 dim_to_delete, in DeleteDimension() function in xla::ShapeUtil1433 shape = DeleteDimension(dim, shape); in FilterDimensions()
218 Shape shape = ShapeUtil::DeleteDimension(dim, hlo->shape()); in AddReduce()