Browse Source

[MS][LITE] fix bug of arm cpu int8 op: conv depthwise int8

tags/v0.7.0-beta
yangruoqi713 6 years ago
parent
commit
6bf9732369
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      mindspore/lite/src/runtime/kernel/arm/fp16/convolution_depthwise_fp16.cc

+ 1
- 1
mindspore/lite/src/runtime/kernel/arm/fp16/convolution_depthwise_fp16.cc View File

@@ -112,7 +112,7 @@ int ConvolutionDepthwiseFp16CPUKernel::Init() {
}
// init sliding_ window param
sliding_ = new SlidingWindowParam;
InitSlidingParam(sliding_, conv_param_, C8NUM);
InitSlidingParamConvDw(sliding_, conv_param_, C8NUM);

ret = InitWeightBias();
if (ret != 0) {


Loading…
Cancel
Save