Searched defs:dims_to_sum_v (Results 1 – 2 of 2) sorted by relevance
552 auto dims_to_sum_v = dims_to_sum.vec(); in _sparse_sum_backward_cuda() local
1647 auto dims_to_sum_v = dims_to_sum.vec(); in _sparse_sum() local1771 auto dims_to_sum_v = dims_to_sum.vec(); in _sparse_sum_backward_cpu() local