Searched refs:DeleteDimension (Results 1 – 12 of 12) sorted by relevance
104 new_input_shape.DeleteDimension(dim); in ExpandInstruction()109 new_kernel_shape.DeleteDimension(dim); in ExpandInstruction()113 new_output_shape.DeleteDimension(dim); in ExpandInstruction()
65 ShapeUtil::DeleteDimension(index_vector_dim, scatter_indices->shape()); in CanonicalizeScatterIndices()
459 ShapeUtil::DeleteDimension(kernel_input_feature_dim, filter->shape()); in HandleConvolution()
1362 new_shape.DeleteDimension(concatenate_dimension); in HandleConcatenate()3428 broadcast_shape1.DeleteDimension(dimension_index); in HandlePad()3429 pad_shape1.DeleteDimension(dimension_index); in HandlePad()3433 broadcast_shape1.DeleteDimension(dimension_index); in HandlePad()3434 pad_shape1.DeleteDimension(dimension_index); in HandlePad()
97 void Shape::DeleteDimension(int64 dim_to_delete) { in DeleteDimension() function in xla::Shape
99 void DeleteDimension(int64 dim_to_delete);
1568 /* static */ Shape ShapeUtil::DeleteDimension(int64 dim_to_delete, in DeleteDimension() function in xla::ShapeUtil1571 shape.DeleteDimension(dim_to_delete); in DeleteDimension()1628 shape = DeleteDimension(dim, shape); in FilterDimensions()
685 static Shape DeleteDimension(int64 dim_to_delete, Shape shape);
210 ShapeUtil::DeleteDimension(reduced_input_dimension, input_shape); in RewriteBatchDimensionLargerThanTile()
109 new_shape.DeleteDimension(input_feature_dimension); in ConvertBatchGroupedToFeatureGroupedConvolution()601 new_shape.DeleteDimension(output_feature_dimension); in MatchBackwardInput()
86 zero_size_shape.DeleteDimension(n_dims - 1); in GetMatrixDiagonal()337 shape->DeleteDimension(dim); in DeleteDimsFromContainer()
309 list_shape.DeleteDimension(0); in Compile()