1 #include "caffe2/experiments/operators/tt_pad_op.h" 6 REGISTER_CPU_OPERATOR(TTPad, TTPadOp<float, CPUContext>);
7 OPERATOR_SCHEMA(TTPad).NumInputs(1).NumOutputs(2).EnforceInplace({{0, 0}});
9 REGISTER_CPU_OPERATOR(TTPadGradient, TTPadGradientOp<float, CPUContext>);
10 OPERATOR_SCHEMA(TTPadGradient)
13 .EnforceInplace({{0, 0}});
15 class GetTTPadGradient :
public GradientMakerBase {
16 using GradientMakerBase::GradientMakerBase;
17 vector<OperatorDef> GetGradientDefs()
override {
18 return SingleGradientDef(
21 vector<string>{GO(0), O(1)},
22 vector<string>{GI(0)},
27 REGISTER_GRADIENT(TTPad, GetTTPadGradient);
A global dictionary that holds information about what Caffe2 modules have been loaded in the current ...