Caffe2 - C++ API
A deep learning, cross platform ML framework
int8_concat_op.cc
1 #include "caffe2/operators/quantized/int8_concat_op.h"
2 
3 #include "caffe2/operators/concat_split_op.h"
4 
5 namespace caffe2 {
6 
7 REGISTER_CPU_OPERATOR(Int8Concat, int8::Int8ConcatOp);
8 
9 OPERATOR_SCHEMA(Int8Concat)
10  .NumInputs(1, INT_MAX)
11  .NumOutputs(1, 2)
12  .Arg("Y_scale", "Output tensor quantization scale")
13  .Arg("Y_zero_point", "Output tensor quantization offset")
14  .Arg("axis", "Which axis to concat on")
15  .Arg(
16  "add_axis",
17  "Pass 1 to add the axis specified in arg 'axis' to all "
18  "input tensors")
19  .TensorInferenceFunction(
20  OpSchema::NeedsAllInputShapes(TensorInferenceForConcat))
21  .CostInferenceFunction(CostInferenceForConcat)
22  .SetDoc("Concatenate a list of tensors into a single tensor")
23  .Output(0, "concat_result", "Concatenated tensor")
24  .Output(1, "split_info", "The dimensions of the inputs.")
25  .InheritOnnxSchema("Concat");
26 
27 } // namespace caffe2
A global dictionary that holds information about what Caffe2 modules have been loaded in the current ...
Definition: blob.h:13
static TensorInferenceFunctionType NeedsAllInputShapes(TensorInferenceFunctionType f)
A wrapper that makes an infer tensor function to return unknown shape for all outputs if any one of t...