projects / platform / upstream / tvm.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: 85a1d3f) | patch
[topi] add ARM v8.2 udot (uint8) support (#3978)
authorYizhi Liu <liuyizhi@apache.org>
Tue, 1 Oct 2019 15:40:16 +0000 (23:40 +0800)
committerGitHub <noreply@github.com>
Tue, 1 Oct 2019 15:40:16 +0000 (23:40 +0800)
commit5cc17649f491299ddf15a8eb144fbb6732382c9a
tree3978fc8cc332d55e8093c96804088dca89fc039dtree | snapshot
parent85a1d3ff8c18682d6f1c1713933ab26380be36abcommit | diff
[topi] add ARM v8.2 udot (uint8) support (#3978)

* [topi] add ARM v8.2 udot (uint8) support

* fix test case

* fix common conv2d schedule

* add back fp32_time in test

* fix lint

* fix doc, add support for int32_lanes=4, signed int

* fix lint

* add ic_bn % 4 checker in schedule
topi/python/topi/arm_cpu/__init__.py diff | blob | history
topi/python/topi/arm_cpu/conv2d_int8.py [new file with mode: 0644] blob
topi/python/topi/arm_cpu/tensor_intrin.py [new file with mode: 0644] blob
topi/python/topi/generic/conv2d.py [new file with mode: 0644] blob
topi/python/topi/nn/conv2d.py diff | blob | history
topi/python/topi/x86/conv2d_avx_1x1.py diff | blob | history
topi/python/topi/x86/conv2d_avx_common.py diff | blob | history
topi/python/topi/x86/conv2d_int8.py diff | blob | history
topi/recipe/conv/test_conv_int8_arm.py [new file with mode: 0644] blob
Domain: Machine Learning / ML Framework;