Searched refs:src_single_zero (Results 1 – 1 of 1) sorted by relevance
6845 Tensor src_single_zero = bitwise_and(src_zero, src_num_zeros == 1); in scatter_reduce_backward() local6849 Tensor masked_src = src.masked_fill(src_single_zero, 1); in scatter_reduce_backward()6853 src_single_zero, in scatter_reduce_backward()6940 Tensor src_single_zero = bitwise_and(src_zero, src_num_zeros == 1); in index_reduce_backward() local6944 Tensor masked_src = source.masked_fill(src_single_zero, 1); in index_reduce_backward()6948 src_single_zero, in index_reduce_backward()