Searched refs:output_indices_shape (Results 1 – 1 of 1) sorted by relevance
98 TensorShape output_indices_shape({0, rank}); in Compute() local100 output_indices_shape, in Compute()141 TensorShape output_indices_shape({N_full, rank}); in Compute() local143 output_indices_shape, in Compute()