Home
last modified time | relevance | path

Searched refs:RemoveLastDims (Results 1 – 12 of 12) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dsvd_op_gpu.cu.cc111 shapeRaw.RemoveLastDims(2); in RunSVD()
243 shapeRaw.RemoveLastDims(2); in PerformSVD_MgeqN()
311 shapeRaw.RemoveLastDims(2); in ComputeAsync()
Dgather_nd_op.cc127 result_shape.RemoveLastDims(1); in DoGatherNd()
192 shape.RemoveLastDims(1); in DoGatherNd()
Dself_adjoint_eig_v2_op_gpu.cc67 eigenvalues_shape.RemoveLastDims(1); in ComputeAsync()
Dmatrix_set_diag_op.cc66 expected_diag_shape.RemoveLastDims(2); in Compute()
Dmatrix_solve_op.cc252 transposed_rhs_shape.RemoveLastDims(2); in ComputeAsync()
Dlookup_table_op.cc542 expected_value_shape.RemoveLastDims(key_shape.dims()); in CheckKeyAndValueTensorsForImport()
804 output_shape.RemoveLastDims(table->key_shape().dims()); in Compute()
Dscatter_nd_op.cc727 slice_shape.RemoveLastDims(1); in DoScatterNd()
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dmatrix_set_diag_op.cc47 batch_shape.RemoveLastDims(2); in Compile()
Dmatrix_band_part_op.cc58 batch_shape.RemoveLastDims(2); in Compile()
Dgather_op.cc70 indices_shape_no_index_vectors.RemoveLastDims(1); in XlaGather()
/external/tensorflow/tensorflow/core/framework/
Dtensor_shape_test.cc73 TEST(TensorShapeTest, RemoveLastDims) { in TEST() argument
75 s.RemoveLastDims(1); in TEST()
80 s.RemoveLastDims(2); in TEST()
Dtensor_shape.h213 void RemoveLastDims(int n) { in RemoveLastDims() function