mirror of
https://github.com/saymrwulf/pytorch.git
synced 2026-05-15 21:00:47 +00:00
3c2462cf24
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/14163 Some of the names we were using to guard the header file was too short (e.g. DYNAMIC_HISTOGRAM_H). Reviewed By: csummersea Differential Revision: D13115451 fbshipit-source-id: cef8c84c62922616ceea17effff7bdf8d67302a2
20 lines
478 B
C++
20 lines
478 B
C++
#pragma once
|
|
|
|
#include "caffe2/core/operator.h"
|
|
#include "caffe2/quantization/server/dnnlowp.h"
|
|
|
|
namespace caffe2 {
|
|
|
|
template <typename T>
|
|
class DequantizeDNNLowPOp final : public Operator<CPUContext> {
|
|
public:
|
|
USE_OPERATOR_FUNCTIONS(CPUContext);
|
|
DequantizeDNNLowPOp(const OperatorDef& operator_def, Workspace* ws);
|
|
|
|
bool RunOnDevice() override;
|
|
|
|
private:
|
|
std::unique_ptr<dnnlowp::QuantizationFactory> qfactory_;
|
|
}; // class DequantizeDNNLowPOp
|
|
|
|
} // namespace caffe2
|