Skip to content

Instantly share code, notes, and snippets.

@argman
argman / onnx_util.cpp
Created February 25, 2021 09:48
onnx c++ code to extract a sub-graph
#include "onnx_util.h"
void Extractor::prepare() {
for(int i=0; i < mModel.graph().initializer_size(); ++i) {
auto initializerI = mModel.graph().initializer(i);
spdlog::info("Initializer {} : name={}", i, initializerI.name());
wmap[initializerI.name()] = initializerI;
}
for(int i=0; i < mModel.graph().value_info_size(); ++i) {