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

Method Cumsum

src/opr/impl/misc.cpp:158–163  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

156MGB_DYN_TYPE_OBJ_FINAL_IMPL(Cumsum);
157
158Cumsum::Cumsum(VarNode* opr, const Param& param, const OperatorNodeConfig& config)
159 : Super{opr->owner_graph(), config, "Cumsum", {opr}} {
160 init_megdnn_opr(*this, param);
161 add_input({opr}, AddInputSortType::CUR_ADDED);
162 output(0)->add_flag(VarNode::Flag::ALLOW_EMPTY_SHAPE);
163}
164
165#if MGB_ENABLE_GRAD
166MGB_IMPL_OPR_GRAD(Cumsum) {

Callers 1

cumsumFunction · 0.80

Calls 1

owner_graphMethod · 0.45

Tested by

no test coverage detected