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

Function is_opr_memforward_var

src/serialization/impl/serializer.cpp:8–20  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

6
7namespace {
8bool is_opr_memforward_var(mgb::VarNode* var) {
9 if (var) {
10 auto opr = var->owner_opr();
11 if (opr->try_cast_final<mgb::opr::Reshape>() ||
12 opr->try_cast_final<mgb::opr::Broadcast>() ||
13 opr->try_cast_final<mgb::opr::Subtensor>() ||
14 opr->try_cast_final<mgb::opr::AxisAddRemove>() ||
15 opr->try_cast_final<mgb::opr::Dimshuffle>()) {
16 return true;
17 }
18 };
19 return false;
20}
21} // namespace
22
23namespace mgb {

Callers 1

graph_compile_aheadMethod · 0.85

Calls 1

owner_oprMethod · 0.45

Tested by

no test coverage detected