Searched refs:need_to_sort (Results 1 – 1 of 1) sorted by relevance
41 int64 output_nnz, int rank, int concat_dim, bool need_to_sort, in SparseConcatKernel() argument66 if (!need_to_sort) { in SparseConcatKernel()73 if (!need_to_sort) { in SparseConcatKernel()161 const bool need_to_sort = concat_dim != 0; in operator ()() local165 if (need_to_sort) { in operator ()()181 if (need_to_sort) { in operator ()()202 rank, concat_dim, need_to_sort, ind_ptrs.data(), in operator ()()204 (need_to_sort ? output_shape.data() in operator ()()208 if (!need_to_sort) return; in operator ()()