projects / platform / upstream / tvm.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: 40f1886) | patch
add rocm schedules to topi C++ (#4507)
authorThomas Viehmann <tv.code@beamnet.de>
Thu, 12 Dec 2019 01:23:00 +0000 (02:23 +0100)
committermasahi <masahi129@gmail.com>
Thu, 12 Dec 2019 01:22:59 +0000 (10:22 +0900)
commitfd6560e1772e30652d89e69724caecae815aadd2
tree4896442690323556cfe943de15f3cee79de80579tree | snapshot
parent40f1886c62e1c68b7658532700596ff8f6ee26a8commit | diff
add rocm schedules to topi C++ (#4507)

This imports the CUDA schedules to rocm.
topi/include/topi/rocm/injective.h [new file with mode: 0644] blob
topi/include/topi/rocm/pooling.h [new file with mode: 0644] blob
topi/include/topi/rocm/reduction.h [new file with mode: 0644] blob
topi/include/topi/rocm/softmax.h [new file with mode: 0644] blob
topi/src/topi.cc diff | blob | history
Domain: Machine Learning / ML Framework;