saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-07-30 20:18:08 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Call optimised version of depthwise ConvLayer (#3664)

Browse source 
* Call optimised version of depthwise ConvLayer

* Update if statements
This commit is contained in:
Prabhat 2020-04-27 17:41:33 +05:30 committed by GitHub
parent c23b484275
commit d901640817
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
onnxruntime/core/providers/acl/nn/conv.cc
View file

@ -208,7 +208,7 @@ Status Conv<T>::Compute(OpKernelContext* context) const {
if(optimizable) {
//optimized depthwise convolution
#if defined(ACL_1902) || defined(ACL_1905)
auto layer = std::make_shared<arm_compute::NEDepthwiseConvolutionLayer>();
auto layer = std::make_shared<arm_compute::NEDepthwiseConvolutionLayer3x3>();
#endif
#ifdef ACL_1908
auto layer = std::make_shared<arm_compute::NEDepthwiseConvolutionLayerOptimized>();