From 2672cd437fe0018ca6f110ecde9d20b5cc7ef6dc Mon Sep 17 00:00:00 2001 From: nihuini Date: Wed, 13 Mar 2019 14:55:58 +0800 Subject: [PATCH] add layer type index member --- src/layer.cpp | 4 +++- src/layer.h | 2 ++ 2 files changed, 5 insertions(+), 1 deletion(-) 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;