Home
last modified time | relevance | path

Searched refs:src_num_zeros (Results 1 – 1 of 1) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/autograd/
H A DFunctionsManual.cpp6841 Tensor src_num_zeros = in scatter_reduce_backward() local
6845 Tensor src_single_zero = bitwise_and(src_zero, src_num_zeros == 1); in scatter_reduce_backward()
6858 if (GradMode::is_enabled() && (src_num_zeros > 1).any().item<bool>()) { in scatter_reduce_backward()
6937 Tensor src_num_zeros = zeros_like(self) in index_reduce_backward() local
6940 Tensor src_single_zero = bitwise_and(src_zero, src_num_zeros == 1); in index_reduce_backward()
6954 if (GradMode::is_enabled() && (src_num_zeros > 1).any().item<bool>()) { in index_reduce_backward()