Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

add Depthwise_conv support #9

Merged
merged 28 commits into from
Dec 14, 2020
Merged

add Depthwise_conv support #9

merged 28 commits into from
Dec 14, 2020

Conversation

QingChuanWS
Copy link
Contributor

增加depthwise conv算子, 并成功通过test_depthwise测试和tm_classification测试.
测试结果:

Image file not specified, use images/cat.jpg by default.
AutoKernel plugin inited
function:autokernel_plugin_init executed
tengine library version: 1.0-dev
Number Thread  : [1]

Number Cluster : [3]

runing AutoKernel im2col_conv ...
using halide depthwise conv...
...
runing AutoKernel im2col_conv ...
[INFO]: runing Autokernel halide_avgpool...
runing AutoKernel im2col_conv ...
current 99.199 ms

Model name : mobilenet
tengine model file : models/mobilenet.tmfile
label file : models/synset_words.txt
image file : images/cat.jpg
img_h, imag_w, scale, mean[3] : 224 224 0.017 104.007 116.669 122.679

Repeat 1 times, avg time per run is 99.199 ms
max time is 99.199 ms, min time is 99.199 ms
--------------------------------------
8.5741 - "n02123159 tiger cat"
7.8801 - "n02119022 red fox, Vulpes vulpes"
7.8126 - "n02119789 kit fox, Vulpes macrotis"
7.2865 - "n02113023 Pembroke, Pembroke Welsh corgi"
6.3575 - "n02123045 tabby, tabby cat"
--------------------------------------
ALL TEST DONE

@lyuchuny3 lyuchuny3 merged commit dced7b2 into OAID:main Dec 14, 2020
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants