Searched defs:dims_to_sum_b (Results 1 – 2 of 2) sorted by relevance
551 auto dims_to_sum_b = dim_list_to_bitset(dims_to_sum, input_dim); in _sparse_sum_backward_cuda() local
1646 auto dims_to_sum_b = dim_list_to_bitset(dims_to_sum, input_dim); in _sparse_sum() local1770 auto dims_to_sum_b = dim_list_to_bitset(dims_to_sum, input_dim); in _sparse_sum_backward_cpu() local