Function make_graph

src/netron_output.cpp:220–261

}

value make_graph(const module* mod)
{
    value graph   = {{"node", value({})},
                     {"initializer", value({})},
                     {"input", value({})},
                     {"output", value({})},
                     {"valueInfo", value({})}};
    auto ins_uids = make_ins_uids(*mod);
    for(auto ins = mod->begin(); ins != mod->end(); ++ins)
    {
        const auto& name = ins->name();
        if(name == "@literal")
        {
            graph["initializer"].push_back(make_onnx_json_literal(ins, ins_uids));
        }
        else if(name == "@param")
        {
            graph["input"].push_back(make_onnx_json_in_out(ins, ins_uids));
        }
        else if(name == "@return")
        {
            graph["output"].push_back(make_onnx_json_in_out(ins, ins_uids));
        }
        else if(name.find("hip::hip_allocate_memory") != std::string::npos)
        {
            continue;
        }
        else
        {
            graph["node"].push_back(make_onnx_json_node(ins, ins_uids));
            const auto& outputs = ins->outputs();
            for(auto out_ins : outputs)
            {
                if(out_ins->name() != "@return")
                {
                    graph["valueInfo"].push_back(make_onnx_json_edge(ins, out_ins, ins_uids));
                }
            }
        }
    }
    return graph;
}

} // namespace