Searched defs:row_map (Results 1 – 3 of 3) sorted by relevance
524 index_t* row_map, in reduce_crow_indices_dim1_cuda_kernel()543 const index_t* row_map, in reduce_sparse_csr_dim1_cuda_kernel()576 Tensor row_map = at::empty({nrows}, ioptions); in reduce_sparse_csr_dim1_cuda_template() local
139 const RowMap& row_map() const { return row_map_; } in row_map() function
1159 Tensor row_map = at::empty({nrows}, ioptions); in reduce_sparse_csr_dim1_cpu_template() local