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

Method get_opr_comp_node_set

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

Source from the content-addressed store, hash-verified

10/* =================== global functions =================== */
11
12CompNode::UnorderedSet cg::get_opr_comp_node_set(OperatorNodeBase* opr) {
13 CompNode::UnorderedSet rst;
14 for (auto i : opr->output())
15 rst.insert(i->comp_node());
16 if (opr->node_prop().contain(OperatorNodeBase::NodeProp::Flag::SINGLE_COMP_NODE))
17 mgb_assert(rst.size() == 1);
18 return rst;
19}
20
21bool cg::is_all_input_static_storage(OperatorNodeBase* opr) {
22 for (auto&& i : opr->node_prop().dep_map())

Callers

nothing calls this directly

Calls 5

outputMethod · 0.45
insertMethod · 0.45
comp_nodeMethod · 0.45
containMethod · 0.45
sizeMethod · 0.45

Tested by

no test coverage detected