1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
|
#include "int8_given_tensor_fill_op.h"
namespace caffe2 {
OPERATOR_SCHEMA(Int8GivenTensorFill)
.NumInputs(0)
.NumOutputs(1)
.Arg("values", "Input array of type char(byte)")
.Arg("shape", "Input tensor shape")
.Arg("Y_scale", "Output tensor quantization scale")
.Arg("Y_zero_point", "Output tensor quantization offset")
.SetDoc(R"DOC(
Creates quantized tensor of type char(byte) with scale and zero point info.
)DOC")
.Output(0, "Tensor", "An Int8TensorCPU with scale and zero point info")
.TensorInferenceFunction(FillerTensorInference<>);
OPERATOR_SCHEMA(Int8GivenIntTensorFill)
.NumInputs(0)
.NumOutputs(1)
.Arg("values", "Input array of type int32")
.Arg("shape", "Input tensor shape")
.Arg("Y_scale", "Output tensor quantization scale")
.Arg("Y_zero_point", "Output tensor quantization offset")
.SetDoc(R"DOC(
Creates quantized tensor of type int32 with scale and zero point info.
)DOC")
.Output(0, "Tensor", "An Int8TensorCPU with scale and zero point info")
.TensorInferenceFunction(FillerTensorInference<>);
REGISTER_CPU_OPERATOR(Int8GivenTensorFill, int8::Int8GivenTensorFillOp);
REGISTER_CPU_OPERATOR(Int8GivenIntTensorFill, int8::Int8GivenIntTensorFillOp);
} // namespace caffe2
|
