Searched defs:sparse_dims_to_sum_size (Results 1 – 2 of 2) sorted by relevance
562 int64_t sparse_dims_to_sum_size = 0; in _sparse_sum_backward_cuda() local
1665 const int64_t sparse_dims_to_sum_size = dims_to_sum_v.size() - dense_dims_to_sum_v.size(); in _sparse_sum() local1781 int64_t sparse_dims_to_sum_size = 0; in _sparse_sum_backward_cpu() local