Searched defs:maskPrefixSum (Results 1 – 2 of 2) sorted by relevance
78 auto maskPrefixSum = at::empty(self.sizes(), mask.options().dtype(kLong)); in masked_scatter__cuda() local
377 const TensorBase &maskPrefixSum, const TensorBase &source) { in launch_masked_scatter_kernel()