MCPcopy Create free account
hub / github.com/MegEngine/MegEngine / AddUpdate>

Method AddUpdate>

src/plugin/impl/opr_footprint.cpp:61–65  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

59// AddUpdate
60template <>
61uint64_t opr_footprint_func<opr::AddUpdate>(cg::OperatorNodeBase* opr) {
62 mgb_assert(opr->input().size() == 2, "AddUpdate opr should have two inputs");
63 auto&& out_shape = opr->output()[0]->shape();
64 return out_shape.total_nr_elems() * 3;
65}
66
67template <class Conv>
68uint64_t eval_conv_computation(

Callers

nothing calls this directly

Calls 5

sizeMethod · 0.45
inputMethod · 0.45
shapeMethod · 0.45
outputMethod · 0.45
total_nr_elemsMethod · 0.45

Tested by

no test coverage detected