| #include "caffe2/operators/zero_gradient_op.h" |
| |
| namespace caffe2 { |
| |
| REGISTER_CPU_OPERATOR(ZeroGradient, ZeroGradientOp<CPUContext>); |
| OPERATOR_SCHEMA(ZeroGradient) |
| .NumInputs(1) |
| .NumOutputs(0) |
| .SetDoc(R"DOC( |
| ZeroGradient operators doesn't produce any output blobs. One can use |
| this operator to produce 0 gradient for the input blob. |
| )DOC"); |
| |
| struct GetZeroGradientOpGradient : public GradientMakerBase { |
| using GradientMakerBase::GradientMakerBase; |
| std::vector<OperatorDef> GetGradientDefs() override { |
| return SingleGradientDef( |
| "ConstantFill", |
| "", |
| vector<string>{I(0)}, |
| vector<string>{GI(0)}, |
| vector<Argument>{MakeArgument<float>("value", 0.0)}); |
| } |
| }; |
| |
| REGISTER_GRADIENT(ZeroGradient, GetZeroGradientOpGradient); |
| |
| } // namespace caffe2 |
