// // ZeroGrad.cpp // MNN // // Created by MNN on 2019/04/22. // Copyright © 2018, Alibaba Group Holding Limited // #include "ReluGrad.hpp" #include "core/Macro.h" using namespace std; namespace MNN { class ZeroGrad : public OpGrad { public: ZeroGrad() { mType = LINEAR; } virtual std::vector onGrad(Express::EXPRP expr, const std::vector& backwardOutput) override { std::vector result(1, nullptr); result[0] = backwardOutput[0]; return result; } }; static void _create() { static ZeroGrad _c; OpGrad::insert(OpType_ZeroGrad, &_c); } REGISTER_GRAD(ZeroGrad_cpp, _create); };