Searched refs:UpdateGrad (Results 1 – 3 of 3) sorted by relevance
| /dports/misc/tvm/incubator-tvm-0.6.1/src/relay/pass/ |
| H A D | gradient.cc | 364 void UpdateGrad(const Type& t, const Expr& arg, const Expr& grad, LetList* ll) { in UpdateGrad() function
371 UpdateGrad(tt->fields[i], in UpdateGrad()
462 UpdateGrad(call->args[i]->checked_type(), args[i], rev[i], ll); in VisitExpr_()
|
| /dports/misc/py-tvm/incubator-tvm-0.6.1/src/relay/pass/ |
| H A D | gradient.cc | 364 void UpdateGrad(const Type& t, const Expr& arg, const Expr& grad, LetList* ll) { in UpdateGrad() function
371 UpdateGrad(tt->fields[i], in UpdateGrad()
462 UpdateGrad(call->args[i]->checked_type(), args[i], rev[i], ll); in VisitExpr_()
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/src/relay/transforms/ |
| H A D | gradient.cc | 432 void UpdateGrad(const Type& t, const Expr& arg, const Expr& grad, LetList* ll) { in UpdateGrad() function
437 UpdateGrad(tt->fields[i], ll->Push(GetField(arg, i)), ll->Push(GetField(grad, i)), ll); in UpdateGrad()
541 UpdateGrad(call->args[i]->checked_type(), args[i], rev[i], ll); in VisitExpr_()
|