Name |
Last commit
|
Last update |
---|---|---|
.. | ||
conv_forward.cc | ||
cudnn_utils.cc | ||
cudnn_utils.h |
* [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
Name |
Last commit
|
Last update |
---|---|---|
.. | ||
conv_forward.cc | Loading commit data... | |
cudnn_utils.cc | Loading commit data... | |
cudnn_utils.h | Loading commit data... |