Expose layer_to_index in c-api (#4860)

Tencent · Jul 16, 2023 · 411a098 · 411a098
1 parent a9a7be0
commit 411a098
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/src/c_api.cpp b/src/c_api.cpp
@@ -1056,6 +1056,11 @@ ncnn_layer_t ncnn_layer_create_by_type(const char* type)
     layer->forward_inplace_n = __ncnn_layer_forward_inplace_n;
     return layer;
 }
+
+int ncnn_layer_type_to_index(const char* type)
+{
+    return ncnn::layer_to_index(type);
+}
 #endif /* NCNN_STRING */
 
 void ncnn_layer_destroy(ncnn_layer_t layer)

diff --git a/src/c_api.h b/src/c_api.h
@@ -210,6 +210,7 @@ NCNN_EXPORT ncnn_layer_t ncnn_layer_create();
 NCNN_EXPORT ncnn_layer_t ncnn_layer_create_by_typeindex(int typeindex);
 #if NCNN_STRING
 NCNN_EXPORT ncnn_layer_t ncnn_layer_create_by_type(const char* type);
+NCNN_EXPORT int ncnn_layer_type_to_index(const char* type);
 #endif /* NCNN_STRING */
 NCNN_EXPORT void ncnn_layer_destroy(ncnn_layer_t layer);