1 #pragma once 2 #include <c10/core/ScalarType.h> 3 #include <cstdint> 4 5 namespace at { 6 struct TensorIteratorBase; 7 class TensorBase; 8 } 9 10 namespace at { 11 namespace native { 12 /// @param maskPrefixSum[in,out] 13 void launch_masked_scatter_kernel( 14 const TensorBase &self, const TensorBase &mask, 15 const TensorBase &maskPrefixSum, const TensorBase &source); 16 }} 17