[TOPI][Winograd] Optimization of Conv2d Winograd algorithm on Tensor Core (#5485)
authorwsl-inspur <61525780+wsl-inspur@users.noreply.github.com>
Thu, 7 May 2020 10:46:45 +0000 (18:46 +0800)
committerGitHub <noreply@github.com>
Thu, 7 May 2020 10:46:45 +0000 (18:46 +0800)
commite40b8bc9e1229d8ee9d0228d5dbdc2338fd8677d
tree13c121d03a9a3376a1592fe2840507a56b1323c0
parentb3730e575ac0ab4a54fe6984bde35408e93b5e6f
[TOPI][Winograd] Optimization of Conv2d Winograd algorithm on Tensor Core (#5485)
python/tvm/relay/op/strategy/cuda.py
topi/python/topi/cuda/__init__.py
topi/python/topi/cuda/conv2d_alter_op.py
topi/python/topi/cuda/conv2d_nhwc_winograd.py [new file with mode: 0644]
topi/tests/python/test_topi_conv2d_nhwc_winograd.py [new file with mode: 0644]