MCPcopy Create free account
hub / github.com/MegEngine/MegEngine / grad

Function grad

src/opr/impl/loop/grad.cpp:337–342  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

335 void scn_do_execute() override {}
336
337 static VarNode* grad(
338 OperatorNodeBase* opr, size_t wrt_idx, const VarNodeArray& out_grad) {
339 MGB_MARK_USED_VAR(out_grad);
340 auto&& info = opr->cast_final_safe<GradProxy>().m_grad.at(wrt_idx);
341 return info.grad;
342 }
343
344 class _RegGrad {
345 static _RegGrad ins;

Callers 15

test_ShuffleRNGFunction · 0.85
test_tqtFunction · 0.85
runFunction · 0.85
test_lsqFunction · 0.85
workerFunction · 0.85
test_gradFunction · 0.85
test_grad_2Function · 0.85
test_2nd_gradFunction · 0.85
test_wrt_visibilityFunction · 0.85
test_grad_inplaceFunction · 0.85

Calls 1

atMethod · 0.45

Tested by 15

test_ShuffleRNGFunction · 0.68
test_tqtFunction · 0.68
runFunction · 0.68
test_lsqFunction · 0.68
workerFunction · 0.68
test_gradFunction · 0.68
test_grad_2Function · 0.68
test_2nd_gradFunction · 0.68
test_wrt_visibilityFunction · 0.68
test_grad_inplaceFunction · 0.68