Searched defs:masked_self (Results 1 – 1 of 1) sorted by relevance
6836 Tensor masked_self = self.masked_fill(self == 0, 1); in scatter_reduce_backward() local6932 Tensor masked_self = self.masked_fill(self == 0, 1); in index_reduce_backward() local