MCPcopy Create free account
hub / github.com/alibaba/MNN / insert

Method insert

tools/train/source/grad/OpGrad.cpp:30–33  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

28}
29
30void OpGrad::insert(int type, OpGrad* converter) {
31 auto& converterMap = getConverter();
32 converterMap.insert(std::make_pair(type, converter));
33}
34
35std::vector<Express::VARP> OpGrad::gradLinear(Express::VARP loss, const std::vector<Express::VARP>& parameters, const std::vector<Express::VARP>& outputDiff, const std::vector<std::string> blockExpr) {
36 std::map<EXPRP, std::vector<VARP>> backwardMap;

Callers 15

__build_graphMethod · 0.45
__get_dom_pathMethod · 0.45
CalibrationMethod · 0.45
_insertScaleMethod · 0.45
preprocessInputMethod · 0.45
_writeJsonToFlatbufferFunction · 0.45
mainFunction · 0.45
mainFunction · 0.45
gradLinearMethod · 0.45
gradCommonMethod · 0.45
_gradForCommandMatMulMethod · 0.45
_gradForCommandMethod · 0.45

Calls

no outgoing calls

Tested by 2

__build_graphMethod · 0.36
__get_dom_pathMethod · 0.36