Skip to content

Commit

Permalink
[RUNTIME] Add cudnn conv3d (apache#4418)
Browse files Browse the repository at this point in the history
* [RUNTIME] Add cudnn conv3d

* add output checking to test_cudnn.verify()

* fix tests failure

* revised per as review comments

* unify conv_output_shape, conv_find_algo and conv_forward

* convert python list to tvm.array in conv_forward

* revise per as comments

* 'pass as reference' for vector args

* add back con2d/3d seperated implementation

* remove unused included header

* remove extra std::vectors

* remove unused header
  • Loading branch information
optima2005 authored and Xingyu Zhou committed Dec 13, 2019
1 parent 2ab329d commit 0e8a0aa
Show file tree
Hide file tree
Showing 7 changed files with 674 additions and 359 deletions.
Loading

0 comments on commit 0e8a0aa

Please sign in to comment.