|
|
|
@@ -138,7 +138,7 @@ class HybridModel { |
|
|
|
std::map<std::string, NodePtr> device_variable_nodes_; //lint !e148 |
|
|
|
std::map<std::string, NodePtr> host_variable_nodes_; //lint !e148 |
|
|
|
std::map<std::string, std::unique_ptr<TensorValue>> variable_tensors_; |
|
|
|
std::map<std::string, std::unique_ptr<TensorValue>> constant_tensors_; |
|
|
|
std::map<std::string, std::unique_ptr<TensorValue>> constant_tensors_; //lint !e148 |
|
|
|
std::map<NodePtr, std::vector<domi::TaskDef>> task_defs_; |
|
|
|
std::map<NodePtr, GeModelPtr> known_shape_sub_models_; |
|
|
|
|
|
|
|
|