HomeSort by relevance Sort by last modified time
    Searched refs:RemoveLastDims (Results 1 - 12 of 12) sorted by null

  /external/tensorflow/tensorflow/compiler/tf2xla/kernels/
matrix_set_diag_op.cc 47 batch_shape.RemoveLastDims(2);
matrix_band_part_op.cc 58 batch_shape.RemoveLastDims(2);
gather_op.cc 70 indices_shape_no_index_vectors.RemoveLastDims(1);
  /external/tensorflow/tensorflow/core/kernels/
gather_nd_op.cc 127 result_shape.RemoveLastDims(1);
192 shape.RemoveLastDims(1);
svd_op_gpu.cu.cc 111 shapeRaw.RemoveLastDims(2);
243 shapeRaw.RemoveLastDims(2);
311 shapeRaw.RemoveLastDims(2);
matrix_set_diag_op.cc 66 expected_diag_shape.RemoveLastDims(2);
self_adjoint_eig_v2_op_gpu.cc 67 eigenvalues_shape.RemoveLastDims(1);
matrix_solve_op.cc 252 transposed_rhs_shape.RemoveLastDims(2);
lookup_table_op.cc 542 expected_value_shape.RemoveLastDims(key_shape.dims());
804 output_shape.RemoveLastDims(table->key_shape().dims());
    [all...]
scatter_nd_op.cc 727 slice_shape.RemoveLastDims(1);
    [all...]
  /external/tensorflow/tensorflow/core/framework/
tensor_shape_test.cc 73 TEST(TensorShapeTest, RemoveLastDims) {
75 s.RemoveLastDims(1);
80 s.RemoveLastDims(2);
tensor_shape.h 213 void RemoveLastDims(int n) {

Completed in 898 milliseconds