| 4 | using namespace lar; |
| 5 | |
| 6 | void OutputDumper::set(mgb::SymbolVarArray& symb_var) { |
| 7 | for (auto&& i : symb_var) { |
| 8 | auto&& var = i.node(); |
| 9 | DumpInfo info; |
| 10 | info.var_info = mgb::cg::dump_var_info({var}); |
| 11 | info.owner_inputs_info = mgb::cg::dump_var_info(var->owner_opr()->input()); |
| 12 | info.id = var->id(); |
| 13 | m_infos.push_back(info); |
| 14 | } |
| 15 | } |
| 16 | |
| 17 | mgb::ComputingGraph::Callback OutputDumper::bind() { |
| 18 | auto& info = m_infos.at(m_bind_id++); |