1 #include "channel_shuffle_op.h" 6 using GradientMakerBase::GradientMakerBase;
7 vector<OperatorDef> GetGradientDefs()
override {
9 def_.type() +
"Gradient",
11 vector<string>{GO(0)},
12 vector<string>{GI(0)});
17 REGISTER_CPU_OPERATOR(
18 ChannelShuffleGradient,
21 OPERATOR_SCHEMA(ChannelShuffle)
22 .IdenticalTypeAndShape()
25 .InheritOnnxSchema(
"ChannelShuffle");
26 OPERATOR_SCHEMA(ChannelShuffleGradient)
27 .IdenticalTypeAndShape()
A global dictionary that holds information about what Caffe2 modules have been loaded in the current ...
static vector< OperatorDef > SingleGradientDef(const Args &...args)
a helper function to allow one to create one single operator def, which is usually the case for many ...