| 99 | } |
| 100 | |
| 101 | flatbuffers::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 | |
| 112 | void GraphDumperOSSV2::init_oprs_to_dump(const SymbolVarArray& endpoints) { |
| 113 | m_oprs_to_dump.clear(); |