Searched defs:output_indices_shape (Results 1 – 1 of 1) sorted by relevance
98 TensorShape output_indices_shape({0, rank}); in Compute() local141 TensorShape output_indices_shape({N_full, rank}); in Compute() local