mirror of
https://github.com/saymrwulf/pytorch.git
synced 2026-05-14 20:57:59 +00:00
04b33b7231
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/9541 Reviewed By: hlu1 Differential Revision: D8882964 fbshipit-source-id: 06d2e0d227ea6a4a8dc5ef1ea9dd1d449c149b47
11 lines
268 B
C++
11 lines
268 B
C++
#include "caffe2/operators/byte_weight_dequant_op.h"
|
|
|
|
#include "caffe2/utils/math.h"
|
|
|
|
namespace caffe2 {
|
|
|
|
REGISTER_CPU_OPERATOR(ByteWeightDequant, ByteWeightDequantOp<CPUContext>);
|
|
|
|
OPERATOR_SCHEMA(ByteWeightDequant).NumInputs(1).NumOutputs(1);
|
|
|
|
} // namespace caffe2
|