Browse Source

add layer type index member

tags/20190320
nihuini 7 years ago
parent
commit
2672cd437f
2 changed files with 5 additions and 1 deletions
  1. +3
    -1
      src/layer.cpp
  2. +2
    -0
      src/layer.h

+ 3
- 1
src/layer.cpp View File

@@ -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

+ 2
- 0
src/layer.h View File

@@ -139,6 +139,8 @@ public:
#endif // NCNN_VULKAN

public:
// layer type index
int typeindex;
#if NCNN_STRING
// layer type name
std::string type;


Loading…
Cancel
Save