Browse Source

concat slice arm fp16sa pack8

tags/20200916
nihui 5 years ago
parent
commit
e7abc5fbd7
4 changed files with 1188 additions and 1072 deletions
  1. +593
    -527
      src/layer/arm/concat_arm.cpp
  2. +0
    -6
      src/layer/arm/concat_arm.h
  3. +595
    -533
      src/layer/arm/slice_arm.cpp
  4. +0
    -6
      src/layer/arm/slice_arm.h

+ 593
- 527
src/layer/arm/concat_arm.cpp
File diff suppressed because it is too large
View File


+ 0
- 6
src/layer/arm/concat_arm.h View File

@@ -24,16 +24,10 @@ class Concat_arm : virtual public Concat
public:
Concat_arm();

virtual int create_pipeline(const Option& opt);
virtual int destroy_pipeline(const Option& opt);

virtual int forward(const std::vector<Mat>& bottom_blobs, std::vector<Mat>& top_blobs, const Option& opt) const;

protected:
int forward_bf16s_fp16s(const std::vector<Mat>& bottom_blobs, std::vector<Mat>& top_blobs, const Option& opt) const;

public:
ncnn::Layer* packing_pack4;
};

} // namespace ncnn


+ 595
- 533
src/layer/arm/slice_arm.cpp
File diff suppressed because it is too large
View File


+ 0
- 6
src/layer/arm/slice_arm.h View File

@@ -24,16 +24,10 @@ class Slice_arm : virtual public Slice
public:
Slice_arm();

virtual int create_pipeline(const Option& opt);
virtual int destroy_pipeline(const Option& opt);

virtual int forward(const std::vector<Mat>& bottom_blobs, std::vector<Mat>& top_blobs, const Option& opt) const;

protected:
int forward_bf16s_fp16s(const std::vector<Mat>& bottom_blobs, std::vector<Mat>& top_blobs, const Option& opt) const;

public:
ncnn::Layer* packing_pack1;
};

} // namespace ncnn


Loading…
Cancel
Save