Add packing for int8 1x1 convolution and support the int8 group convolution on X86...
authorllyfacebook <34827865+llyfacebook@users.noreply.github.com>
Wed, 22 May 2019 16:11:17 +0000 (09:11 -0700)
committerYizhi Liu <liuyizhi@apache.org>
Wed, 22 May 2019 16:11:17 +0000 (09:11 -0700)
commitf7d7fdcd3881beff0a0ad13c6a82b45aa834fd61
treeba6d7eab8f4eef37c5345fe178e8c29ffee3d6e9
parenta798a01bb6c6518e6c4c302edfbd72098b4c27d9
Add packing for int8 1x1 convolution and support the int8 group convolution on X86 (#2991)

* Support the 1x1 int8 conv with NHWC layout and weight packing

fix linter

* fix the memoize issue

* fix the failed nhwc test

* add the schedule for pack to unbreak other tests

* skip avx512 compile

* Support the 1x1 int8 conv with NHWC layout and weight packing

fix linter

* fix the memoize issue

* fix the failed nhwc test

* add the schedule for pack to unbreak other tests

* skip avx512 compile

* Unify the data_layout and kernel_layout relation

* add asf header

* fix the comment

* retrigger the build/test
topi/python/topi/generic/nn.py
topi/python/topi/nn/conv2d.py
topi/python/topi/x86/conv2d.py
topi/python/topi/x86/conv2d_avx_1x1.py
topi/tests/python/test_topi_conv2d_nhwc_pack_int8.py [new file with mode: 0644]
topi/tests/python/test_topi_group_conv2d_NCHWc_int8.py [new file with mode: 0644]