GithubStar/ncnn
1
0
Fork 0
mirror of https://github.com/Tencent/ncnn.git synced 2026-10-27 19:29:54 +08:00
Code Issues Packages Projects Releases Wiki Activity

binaryop bf16s

Browse Source
This commit is contained in:
nihuini
2020-03-25 16:12:58 +08:00
parent 89ef1f0d66
commit 867ff7ae97
3 changed files with 1102 additions and 60 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/layer/arm/binaryop_arm.cpp
+1095 -60
View File
File diff suppressed because it is too large Load Diff
src/layer/arm/binaryop_arm.h
+5
View File
@@ -27,6 +27,11 @@ public:
virtual int forward(const std::vector<Mat>& bottom_blobs, std::vector<Mat>& top_blobs, const Option& opt) const;
virtual int forward_inplace(Mat& bottom_top_blob, const Option& opt) const;
protected:
int forward_bf16s(const std::vector<Mat>& bottom_blobs, std::vector<Mat>& top_blobs, const Option& opt) const;
int forward_inplace_bf16s(Mat& bottom_top_blob, const Option& opt) const;
};
} // namespace ncnn
tests/test_binaryop.cpp
+2
View File
@@ -39,6 +39,7 @@ static int test_binaryop(const ncnn::Mat& _a, const ncnn::Mat& _b, int op_type)
ncnn::Option opt;
opt.num_threads = 1;
opt.use_vulkan_compute = true;
opt.use_int8_inference = false;
opt.use_fp16_packed = false;
opt.use_fp16_storage = false;
opt.use_fp16_arithmetic = false;
@@ -78,6 +79,7 @@ static int test_binaryop(const ncnn::Mat& _a, float b, int op_type)
ncnn::Option opt;
opt.num_threads = 1;
opt.use_vulkan_compute = true;
opt.use_int8_inference = false;
opt.use_fp16_packed = false;
opt.use_fp16_storage = false;
opt.use_fp16_arithmetic = false;