Browse Source

feat(dnn/arm): add nchw nchw44 first conv

GitOrigin-RevId: 563f57844f
tags/v0.4.0
Megvii Engine Team Xinran Xu 5 years ago
parent
commit
a323f1a449
3 changed files with 6 additions and 4 deletions
  1. +2
    -2
      dnn/src/fallback/conv_bias/opr_impl.cpp
  2. +2
    -1
      dnn/src/fallback/conv_bias/opr_impl.h
  3. +2
    -1
      dnn/src/fallback/convolution/algos.cpp

+ 2
- 2
dnn/src/fallback/conv_bias/opr_impl.cpp View File

@@ -57,8 +57,8 @@ public:
}
for (size_t oc_tile_size : {24, 48}) {
refhold.emplace_back(new AlgoConv1x1(
static_cast<MatrixMulImpl::AlgoBase*>(algo),
oc_tile_size));
static_cast<MatrixMulImpl::AlgoBase*>(algo),
oc_tile_size));
all_algos.emplace_back(refhold.back().get());
}
#if 0


+ 2
- 1
dnn/src/fallback/conv_bias/opr_impl.h View File

@@ -6,7 +6,8 @@
*
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or
* implied.
*/
#pragma once



+ 2
- 1
dnn/src/fallback/convolution/algos.cpp View File

@@ -6,7 +6,8 @@
*
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or
* implied.
*/

#include "src/fallback/convolution/algos.h"


Loading…
Cancel
Save