提交 b04e0466 编写于 作者: M Megvii Engine Team

feat(megbrain): add alias name to model serialization

GitOrigin-RevId: d47e65047309fa0e2f132bacbe23a0988f7894ef
上级 cf53d9e0
......@@ -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_ =
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册