提交 13c2b233 编写于 作者: M mindspore-ci-bot 提交者: Gitee

!4324 fix int8 conv non-optimize bug

Merge pull request !4324 from fuzhiye/tmp
......@@ -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;
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册