Searched refs:input_sparse_dim (Results 1 – 2 of 2) sorted by relevance
558 const int64_t input_sparse_dim = input.sparse_dim(); in _sparse_sum_backward_cuda() local567 if (d < input_sparse_dim) sparse_dims_to_sum_size ++; in _sparse_sum_backward_cuda()568 else dense_dims_to_sum_v.emplace_back(d + 1 - input_sparse_dim); in _sparse_sum_backward_cuda()571 if (d < input_sparse_dim) sparse_dims_to_keep_v.emplace_back(d); 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()590 …return at::_sparse_coo_tensor_with_dims_and_tensors(input_sparse_dim, input_dense_dim, input_sizes… in _sparse_sum_backward_cuda()666 …return at::_sparse_coo_tensor_with_dims_and_tensors(input_sparse_dim, input_dense_dim, input_sizes… in _sparse_sum_backward_cuda()
1777 const int64_t input_sparse_dim = input.sparse_dim(); in _sparse_sum_backward_cpu() local1786 if (d < input_sparse_dim) sparse_dims_to_sum_size ++; in _sparse_sum_backward_cpu()1787 else dense_dims_to_sum_v.emplace_back(d + 1 - input_sparse_dim); in _sparse_sum_backward_cpu()1790 if (d < input_sparse_dim) sparse_dims_to_keep_v.emplace_back(d); 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()1811 …return at::_sparse_coo_tensor_with_dims_and_tensors(input_sparse_dim, input_dense_dim, input_sizes… in _sparse_sum_backward_cpu()1888 …return at::_sparse_coo_tensor_with_dims_and_tensors(input_sparse_dim, input_dense_dim, input_sizes… in _sparse_sum_backward_cpu()