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
|
#include "caffe2/operators/quantized/int8_reshape_op.h"
namespace caffe2 {
REGISTER_CPU_OPERATOR(Int8Reshape, int8::Int8ReshapeOp);
OPERATOR_SCHEMA(Int8Reshape)
.NumInputs(1, 2)
.NumOutputs(2)
.AllowInplace({{0, 0}})
.SetDoc(R"DOC(
Reshape the input tensor similar to numpy.reshape.
It takes a tensor as input and an optional tensor specifying the new shape.
When the second input is absent, an extra argument `shape` must be specified.
It outputs the reshaped tensor as well as the original shape.
At most one dimension of the new shape can be -1. In this case, the value is
inferred from the size of the tensor and the remaining dimensions. A dimension
could also be 0, in which case the actual dimension value is going to be copied
from the input tensor.
)DOC")
.Arg("shape", "New shape")
.Arg("Y_scale", "Output tensor quantization scale")
.Arg("Y_zero_point", "Output tensor quantization offset")
.Input(0, "data", "An input tensor.")
.Input(1, "new_shape", "New shape.")
.Output(0, "reshaped", "Reshaped data.")
.Output(1, "old_shape", "Original shape.");
} // namespace caffe2
|
