Searched refs:sparse_dims_to_sum_size (Results 1 – 2 of 2) sorted by relevance
1665 const int64_t sparse_dims_to_sum_size = dims_to_sum_v.size() - dense_dims_to_sum_v.size(); in _sparse_sum() local1666 const bool sum_all_sparse_dim = (sparse_dim == sparse_dims_to_sum_size); in _sparse_sum()1686 if (sparse_dims_to_sum_size == 0) { in _sparse_sum()1690 … new_indices = at::empty({sparse_dim - sparse_dims_to_sum_size, input._nnz()}, indices.options()); in _sparse_sum()1781 int64_t sparse_dims_to_sum_size = 0; in _sparse_sum_backward_cpu() local1786 if (d < input_sparse_dim) sparse_dims_to_sum_size ++; in _sparse_sum_backward_cpu()1794 const bool sum_all_sparse_dim = (input_sparse_dim == sparse_dims_to_sum_size); in _sparse_sum_backward_cpu()1796 const bool sum_sparse_dim = (sparse_dims_to_sum_size > 0); in _sparse_sum_backward_cpu()
562 int64_t sparse_dims_to_sum_size = 0; in _sparse_sum_backward_cuda() local567 if (d < input_sparse_dim) sparse_dims_to_sum_size ++; in _sparse_sum_backward_cuda()575 const bool sum_all_sparse_dim = (input_sparse_dim == sparse_dims_to_sum_size); in _sparse_sum_backward_cuda()577 const bool sum_sparse_dim = (sparse_dims_to_sum_size > 0); in _sparse_sum_backward_cuda()