Searched refs:src_num_zeros (Results 1 – 1 of 1) sorted by relevance
6841 Tensor src_num_zeros = in scatter_reduce_backward() local6845 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() local6940 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()