diff --git a/src/net.h b/src/net.h index 10cf307ab..94070422f 100644 --- a/src/net.h +++ b/src/net.h @@ -61,6 +61,7 @@ public: // register custom layer by layer type name // return 0 if success int register_custom_layer(const char* type, layer_creator_func creator, layer_destroyer_func destroyer = 0, void* userdata = 0); + virtual int custom_layer_to_index(const char* type); #endif // NCNN_STRING // register custom layer by layer type // return 0 if success @@ -147,7 +148,6 @@ protected: #if NCNN_STRING int find_blob_index_by_name(const char* name) const; int find_layer_index_by_name(const char* name) const; - virtual int custom_layer_to_index(const char* type); virtual Layer* create_custom_layer(const char* type); #endif // NCNN_STRING virtual Layer* create_custom_layer(int index);