diff --git a/src/layer.cpp b/src/layer.cpp index ae6df4bdb..45084805e 100644 --- a/src/layer.cpp +++ b/src/layer.cpp @@ -223,7 +223,9 @@ Layer* create_layer(int index) if (!layer_creator) return 0; - return layer_creator(); + Layer* layer = layer_creator(); + layer->typeindex = index; + return layer; } } // namespace ncnn diff --git a/src/layer.h b/src/layer.h index 4439f662b..dc9c95013 100644 --- a/src/layer.h +++ b/src/layer.h @@ -139,6 +139,8 @@ public: #endif // NCNN_VULKAN public: + // layer type index + int typeindex; #if NCNN_STRING // layer type name std::string type;