Caffe2 - C++ API
A deep learning, cross platform ML framework
int8_channel_shuffle_op.cc
1 #include "caffe2/operators/quantized/int8_channel_shuffle_op.h"
2 
3 namespace caffe2 {
4 
5 REGISTER_CPU_OPERATOR(Int8ChannelShuffle, int8::Int8ChannelShuffleOp);
6 
7 OPERATOR_SCHEMA(Int8ChannelShuffle)
8  .IdenticalTypeAndShape()
9  .Arg("Y_scale", "Output tensor quantization scale")
10  .Arg("Y_zero_point", "Output tensor quantization offset")
11  .NumInputs(1)
12  .NumOutputs(1);
13 
14 } // namespace caffe2
A global dictionary that holds information about what Caffe2 modules have been loaded in the current ...
Definition: blob.h:13