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

Method build_output_var

src/serialization/impl/serializer_oss_v2.cpp:101–110  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

99}
100
101flatbuffers::Offset<fbs::v2::OutputVar> GraphDumperOSSV2::build_output_var(
102 const SymbolVar var) {
103 auto out_node = var.node();
104 if (m_var2midtensor_id.find(var.node()) == m_var2midtensor_id.end()) {
105 mgb_assert(m_var_remove_in_dump.find(var.node()) != m_var_remove_in_dump.end());
106 out_node = m_var_remove_in_dump[var.node()];
107 }
108 return fbs::v2::CreateOutputVar(
109 m_builder, m_var2midtensor_id.at(out_node), var.node()->id());
110}
111
112void GraphDumperOSSV2::init_oprs_to_dump(const SymbolVarArray& endpoints) {
113 m_oprs_to_dump.clear();

Callers

nothing calls this directly

Calls 5

nodeMethod · 0.45
findMethod · 0.45
endMethod · 0.45
atMethod · 0.45
idMethod · 0.45

Tested by

no test coverage detected