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

Method grad

src/core/impl/graph/helper.cpp:71–75  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

69}
70
71SymbolVar cg::grad(
72 SymbolVar target, SymbolVar wrt, bool warn_mid_wrt,
73 bool return_zero_for_nodep) {
74 return grad(target, SymbolVarArray{wrt}, warn_mid_wrt, return_zero_for_nodep)[0];
75}
76
77SymbolVarArray cg::grad(
78 SymbolVar target_, SymbolVarArray wrts_, bool warn_mid_wrt,

Callers

nothing calls this directly

Calls 13

gradFunction · 0.85
enter_record_modeMethod · 0.80
emplace_backMethod · 0.80
flush_record_oprsMethod · 0.80
clean_cacheMethod · 0.80
nodeMethod · 0.45
reserveMethod · 0.45
sizeMethod · 0.45
owner_graphMethod · 0.45
inputMethod · 0.45
owner_oprMethod · 0.45
push_backMethod · 0.45

Tested by

no test coverage detected