Searched refs:RemoveLastDims (Results 1 – 12 of 12) sorted by relevance
111 shapeRaw.RemoveLastDims(2); in RunSVD()243 shapeRaw.RemoveLastDims(2); in PerformSVD_MgeqN()311 shapeRaw.RemoveLastDims(2); in ComputeAsync()
127 result_shape.RemoveLastDims(1); in DoGatherNd()192 shape.RemoveLastDims(1); in DoGatherNd()
67 eigenvalues_shape.RemoveLastDims(1); in ComputeAsync()
66 expected_diag_shape.RemoveLastDims(2); in Compute()
252 transposed_rhs_shape.RemoveLastDims(2); in ComputeAsync()
542 expected_value_shape.RemoveLastDims(key_shape.dims()); in CheckKeyAndValueTensorsForImport()804 output_shape.RemoveLastDims(table->key_shape().dims()); in Compute()
727 slice_shape.RemoveLastDims(1); in DoScatterNd()
47 batch_shape.RemoveLastDims(2); in Compile()
58 batch_shape.RemoveLastDims(2); in Compile()
70 indices_shape_no_index_vectors.RemoveLastDims(1); in XlaGather()
73 TEST(TensorShapeTest, RemoveLastDims) { in TEST() argument75 s.RemoveLastDims(1); in TEST()80 s.RemoveLastDims(2); in TEST()
213 void RemoveLastDims(int n) { in RemoveLastDims() function