Browse Source

ncnnoptimize tool

tags/20190611
nihuini 7 years ago
parent
commit
7984ffcb4d
3 changed files with 1225 additions and 2 deletions
  1. +1
    -1
      src/layer/reorg.h
  2. +1
    -1
      src/layer/reshape.h
  3. +1223
    -0
      tools/ncnnoptimize.cpp

+ 1
- 1
src/layer/reorg.h View File

@@ -35,7 +35,7 @@ public:
virtual int forward(const VkMat& bottom_blob, VkMat& top_blob, VkCompute& cmd, const Option& opt) const;
#endif // NCNN_VULKAN

private:
public:
int stride;

#if NCNN_VULKAN


+ 1
- 1
src/layer/reshape.h View File

@@ -35,7 +35,7 @@ public:
virtual int forward(const VkMat& bottom_blob, VkMat& top_blob, VkCompute& cmd, const Option& opt) const;
#endif // NCNN_VULKAN

private:
public:
// reshape flag
// 0 = copy from bottom
// -1 = remaining


+ 1223
- 0
tools/ncnnoptimize.cpp
File diff suppressed because it is too large
View File


Loading…
Cancel
Save