From b04e0466bb2cf1a025adaa3e707dbef03429684f Mon Sep 17 00:00:00 2001
From: Megvii Engine Team <megengine@megvii.com>
Date: Wed, 9 Dec 2020 18:20:41 +0800
Subject: [PATCH] feat(megbrain): add alias name to model serialization

GitOrigin-RevId: d47e65047309fa0e2f132bacbe23a0988f7894ef
---
 src/serialization/include/megbrain/serialization/load_dump_config.h | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/src/serialization/include/megbrain/serialization/load_dump_config.h b/src/serialization/include/megbrain/serialization/load_dump_config.h
index da16be41..a4c67c7c 100644
--- a/src/serialization/include/megbrain/serialization/load_dump_config.h
+++ b/src/serialization/include/megbrain/serialization/load_dump_config.h
@@ -51,6 +51,10 @@ struct GraphDumpConfig {
     //! tensor value without layout; useful for compression or encryption
     TensorValueDumper tensor_value_dumper;
 
+    //! a list of output nodes and names. one output node may have multiple
+    //! names. this list record the mapping between output node and it's name
+    std::vector<std::pair<std::string, SymbolVar>> alias_name_map;
+
     GraphDumpConfig(int keep_var_name_ = 1, bool keep_param_name_ = false,
                     bool keep_opr_priority_ = false,
                     const std::shared_ptr<UserDataContainer>& user_data_ =