Browse Source

1.fix int8 conv non-optimize bug

2.Malloc output tensor in runtime for quant cast type op
tags/v0.7.0-beta
fuzhiye 5 years ago
parent
commit
3cda158a84
1 changed files with 3 additions and 1 deletions
  1. +3
    -1
      mindspore/lite/src/runtime/kernel/arm/nnacl/pack.cc

+ 3
- 1
mindspore/lite/src/runtime/kernel/arm/nnacl/pack.cc View File

@@ -74,7 +74,9 @@ void PackWeightInt8(int8_t *weight_data, ConvParameter *conv_param, int8_t *pack

for (int m = 0; m < kernel_plane; m++) {
int kernel_plane_stride = m * in_channel;
int packed_kernel_plane_stride = m * C4NUM;
int plane_block = m / C4NUM;
int plane_res = m % C4NUM;
int packed_kernel_plane_stride = plane_block * C4NUM * C4NUM * ic4 * C4NUM + plane_res * C4NUM;
for (int i = 0; i < ic4; i++) {
int channel_block_stride = kernel_plane_stride + i * C4NUM;
int packed_channel_block_size = packed_kernel_plane_stride + i * C4NUM * C4NUM * C4NUM;


Loading…
Cancel
Save