Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/csrc/autograd/
H A DFunctionsManual.cpp6845 Tensor src_single_zero = bitwise_and(src_zero, src_num_zeros == 1); in scatter_reduce_backward() local
6849 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() local
6944 Tensor masked_src = source.masked_fill(src_single_zero, 1); in index_reduce_backward()
6948 src_single_zero, in index_reduce_backward()