Searched refs:select_sparse_csr_worker (Results 1 – 1 of 1) sorted by relevance
967 Tensor select_sparse_csr_worker(const Tensor& self, int64_t dim, int64_t index) { in select_sparse_csr_worker() function1255 return select_sparse_csr_worker<true, false>(self, dim, index); in select_sparse_csr()1259 return select_sparse_csr_worker<false, true>(self, dim, index); in select_copy_sparse_csr()