1 #include <ATen/ATen.h>
2 #include <ATen/NativeFunctions.h>
3 #include <torch/library.h>
4 #include <ATen/quantized/Quantizer.h>
5 #include <ATen/native/quantized/cpu/QuantizedOps.h>
6
7 namespace at {
8 namespace native {
9
10 DEFINE_DISPATCH(qdropout_stub);
11
quantized_dropout(const Tensor & qx,double output_scale,int64_t output_zero_point,const Scalar & p,bool training)12 static Tensor quantized_dropout(
13 const Tensor& qx, double output_scale, int64_t output_zero_point, const Scalar& p, bool training) {
14 return qx;
15 }
16
TORCH_LIBRARY_IMPL(quantized,QuantizedCPU,m)17 TORCH_LIBRARY_IMPL(quantized, QuantizedCPU, m) {
18 m.impl(TORCH_SELECTIVE_NAME("quantized::dropout"), quantized_dropout);
19 }
20
21 }} // namespace at::native
22