- 03 Dec, 2019 1 commit
-
-
* [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
optima2005 committed
-
- 25 Nov, 2019 1 commit
-
-
* add half and mix precision support to cublas backend * add TensorCore support in CuDNN * enhance CuDNN support * address comments and fix lint * fix * add fp16 test
Siyuan Feng committed
-
- 08 Apr, 2019 1 commit
-
-
* [HEADER] ASF header dir=include * [HEADER] ASF Header dir=src * [HEADER] ASF Header -dir=python * [HEADER] ASF header dir=topi * [HEADER] ASF Header dir=nnvm * [HEADER] ASF Header -dir=tutorials * [HEADER] ASF Header dir=tests * [HEADER] ASF Header -dir=docker * fix whitespace * [HEADER] ASF Header -dir=jvm * [HEADER] ASF Header -dir=web * [HEADER] ASF Header --dir=apps * [HEADER] ASF Header --dir=vta * [HEADER] ASF Header -dir=go * temp * [HEADER] ASF Header --dir=rust * [HEADER] Add ASF Header --dir=cmake * [HEADER] ASF Header --dir=docs * [HEADER] Header for Jenkinsfile * [HEADER] ASF Header to toml and md * [HEADER] ASF Header to gradle * Finalize rat cleanup * Fix permission * Fix java test * temporary remove nnvm onnx test
Tianqi Chen committed
-
- 04 Apr, 2018 1 commit
-
-
Lianmin Zheng committed
-
- 29 Dec, 2017 1 commit
-
-
* use cudnn findalgo to choose the best algo * fix lint
masahi committed
-
- 15 Aug, 2017 1 commit
-
-
* [Contrib] CuDNN v7 Support * Add test
ziheng committed
-