Searched refs:ReverseAD (Results 1 – 3 of 3) sorted by relevance
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/src/relay/transforms/ |
| H A D | gradient.cc | 450 struct ReverseAD : ExprMutator { struct
460 explicit ReverseAD(const Optional<IRModule>& mod, const Var& bp, in ReverseAD() argument
502 ll->Push(ReverseAD(mod, dup_bp, ad_vars, ad_gvars)(DeDup(x))); in VisitCheckpoint()
614 return Function(params, ReverseAD(mod, new_bp, ad_vars, ad_gvars)(op->body), in VisitExpr_()
664 Expr rev = ReverseAD(mod, bp, std::make_shared<ReverseAD::ADVarMap>(), in Gradient()
665 std::make_shared<ReverseAD::ADGlobalVarMap>())(e); in Gradient()
|
| /dports/misc/tvm/incubator-tvm-0.6.1/src/relay/pass/ |
| H A D | gradient.cc | 387 struct ReverseAD : ExprMutator { struct
394 explicit ReverseAD(const Var& bp, std::shared_ptr<ADVarMap> ad_vars) in ReverseAD() function
417 ReverseAD dup_diff(dup_bp, ad_vars); in VisitCheckpoint()
541 Expr rev = ReverseAD(bp, std::make_shared<ReverseAD::ADVarMap>())(e); in Gradient()
|
| /dports/misc/py-tvm/incubator-tvm-0.6.1/src/relay/pass/ |
| H A D | gradient.cc | 387 struct ReverseAD : ExprMutator { struct
394 explicit ReverseAD(const Var& bp, std::shared_ptr<ADVarMap> ad_vars) in ReverseAD() function
417 ReverseAD dup_diff(dup_bp, ad_vars); in VisitCheckpoint()
541 Expr rev = ReverseAD(bp, std::make_shared<ReverseAD::ADVarMap>())(e); in Gradient()
|