Searched refs:res_sparse_impl (Results 1 – 2 of 2) sorted by relevance
414 auto* res_sparse_impl = get_sparse_impl(res); variable415 res_sparse_impl->raw_resize_(res_sparse_dim, res_dense_dim, res_shape);416 res_sparse_impl->set_indices_and_values_unsafe(res_indices, res_values);417 res_sparse_impl->set_nnz_and_narrow(res_nnz);
745 auto* res_sparse_impl = get_sparse_impl(res); in sparse_mask_like_prepare_sparse_inputs() local746 res_sparse_impl->raw_resize_(t.sparse_dim(), t.dense_dim(), t.sizes()); in sparse_mask_like_prepare_sparse_inputs()749 res_sparse_impl->set_indices_and_values_unsafe(res_indices, res_values); in sparse_mask_like_prepare_sparse_inputs()750 res_sparse_impl->set_nnz_and_narrow(t._nnz()); in sparse_mask_like_prepare_sparse_inputs()