OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
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