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