1 #pragma once 2 #include <ATen/native/TensorIterator.h> 3 4 namespace at { 5 namespace native { 6 using masked_fill_kernel_quantized_fn = void(*)(TensorIterator& iter, const Scalar& value, double scale, int zero_point); 7 using index_put_kernel_quantized_fn = void(*)(TensorIterator& iter, IntArrayRef index_size, IntArrayRef index_stride, bool accumulate, double scale, int zero_point); 8 9 DECLARE_DISPATCH(masked_fill_kernel_quantized_fn, masked_fill_kernel_quantized_stub); 10 DECLARE_DISPATCH(index_put_kernel_quantized_fn, index_put_kernel_quantized_stub); 11 12 13 } // native 14 } // at 15