Searched refs:DeleteDimension (Results 1 – 7 of 7) sorted by relevance
359 ShapeUtil::DeleteDimension(kernel_input_feature_dim, filter->shape()); in HandleConvolution()449 ShapeUtil::DeleteDimension(new_spatial_dim, new_convolution->shape()); in HandleConvolution()
65 ShapeUtil::DeleteDimension(index_vector_dim, scatter_indices->shape()); in CanonicalizeScatterIndices()
269 ShapeUtil::DeleteDimension(dim, hlo->shape()), hlo)); in StripDim()
97 void Shape::DeleteDimension(int64 dim_to_delete) { in DeleteDimension() function in xla::Shape
83 void DeleteDimension(int64 dim_to_delete);
1410 /* static */ Shape ShapeUtil::DeleteDimension(int64 dim_to_delete, in DeleteDimension() function in xla::ShapeUtil1413 shape.DeleteDimension(dim_to_delete); in DeleteDimension()1427 shape = DeleteDimension(dim, shape); in FilterDimensions()
609 static Shape DeleteDimension(int64 dim_to_delete, Shape shape);