Home
last modified time | relevance | path

Searched defs:output_indices_shape (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dsparse_fill_empty_rows_op.cc93 TensorShape output_indices_shape; in operator ()() local
151 TensorShape output_indices_shape; in operator ()() local
H A Dsparse_fill_empty_rows_op_gpu.cu.cc432 TensorShape output_indices_shape({N_full, rank}); in AllocateOutputsExceptEmptyRowIndicator() local