Searched defs:row_map (Results 1 – 4 of 4) sorted by relevance
139 const RowMap& row_map() const { return row_map_; } in row_map() function
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
480 auto row_map = table.QueryToRowMap(q); in GetByConstraint() local
1159 Tensor row_map = at::empty({nrows}, ioptions); in reduce_sparse_csr_dim1_cpu_template() local