* [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 |
---|---|---|
.. | ||
test_binutil.py | Loading commit data... | |
test_cblas.py | Loading commit data... | |
test_cublas.py | Loading commit data... | |
test_cudnn.py | Loading commit data... | |
test_dlpack.py | Loading commit data... | |
test_gemm_acc16.py | Loading commit data... | |
test_gemm_acc32_vnni.py | Loading commit data... | |
test_miopen.py | Loading commit data... | |
test_mps.py | Loading commit data... | |
test_mxnet_bridge.py | Loading commit data... | |
test_nnpack.py | Loading commit data... | |
test_random.py | Loading commit data... | |
test_rocblas.py | Loading commit data... | |
test_rpc_proxy.py | Loading commit data... | |
test_rpc_tracker.py | Loading commit data... | |
test_sort.py | Loading commit data... | |
test_sparse.py | Loading commit data... |