[UPD] add generate resource for const folder (#1835)

Co-authored-by: ealinli <[email protected]>
Tencent · Nov 25, 2022 · 97e627e · 97e627e
1 parent 2c849dc
commit 97e627e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/source/tnn/core/default_network.cc b/source/tnn/core/default_network.cc
@@ -201,7 +201,7 @@ Status DefaultNetwork::InitLayers(NetStructure *net_structure, NetResource *net_
             dynamic_cast<ReformatLayerParam *>(layer_info->param.get())->dst_format : input_fmt;
 
 #ifdef GENERATE_RESOURCE
-        if (runtime_model_ == RUNTIME_MODE_NORMAL) {
+        if (runtime_model_ == RUNTIME_MODE_NORMAL || runtime_model_ == RUNTIME_MODE_CONST_FOLD) {
             LayerType type       = layer_info->type;
             BaseLayer *cur_layer = CreateLayer(type);
             if (cur_layer == NULL) {