Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

[RUNTIME] Add cudnn conv3d #4418

Merged
merged 12 commits into from
Dec 3, 2019
Merged

[RUNTIME] Add cudnn conv3d #4418

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
12 commits
Select commit Hold shift + click to select a range
ba0b1fd
[RUNTIME] Add cudnn conv3d
optima2005 Nov 25, 2019
00372e7
add output checking to test_cudnn.verify()
optima2005 Nov 25, 2019
dc1f9cf
fix tests failure
optima2005 Nov 27, 2019
d7aa837
revised per as review comments
optima2005 Nov 27, 2019
0c2b622
unify conv_output_shape, conv_find_algo and conv_forward
optima2005 Nov 28, 2019
983433f
convert python list to tvm.array in conv_forward
optima2005 Nov 29, 2019
85b711f
revise per as comments
optima2005 Nov 29, 2019
aede7e8
'pass as reference' for vector args
optima2005 Nov 29, 2019
e2d022d
add back con2d/3d seperated implementation
optima2005 Nov 30, 2019
1394402
remove unused included header
optima2005 Nov 30, 2019
40ec63b
remove extra std::vectors
optima2005 Dec 3, 2019
ae949a6
remove unused header
optima2005 Dec 3, 2019
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Loading