You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

Pei Yang 994673bf4f change avg pooling and global pooling to trt layer in dynamic shape mode (#28702) ... * change avg pooling and global pooling to trt layer * add support for static shape global pooling * modify trt errmsg		5 years ago
..
CMakeLists.txt	Add cholesky_op (#23543)	5 years ago
cublas.cc	fix cublas warp error	7 years ago
cublas.h	use iwyu clean include (#27267)	5 years ago
cuda_driver.cc	remove WITH_DSO compile option (#25444)	5 years ago
cuda_driver.h	use iwyu clean include (#27267)	5 years ago
cudnn.cc	[bug fix]:Memory increases after adapting the cudnn version to cudnn8 (#27436)	5 years ago
cudnn.h	tune backward filter algorithm for float16 (#27529)	5 years ago
cupti.cc	use iwyu clean include (#27267)	5 years ago
cupti.h	remove WITH_DSO compile option (#25444)	5 years ago
cupti_lib_path.h.in	Fix the compilation on CUDA 9.1/GCC 5.3	7 years ago
curand.cc	Fix the grammar in copyright. (#8403)	7 years ago
curand.h	use iwyu clean include (#27267)	5 years ago
cusolver.cc	Fix cusolver loader for Windows (#24157)	5 years ago
cusolver.h	use iwyu clean include (#27267)	5 years ago
dynamic_loader.cc	fix dynamic_loader more safe and error message on windows (#28117)	5 years ago
dynamic_loader.h	Add cholesky_op (#23543)	5 years ago
mklml.cc	Fix the definition issue when used mkl_scsrmm and mkl_dcsrmm functions. (#19774)	6 years ago
mklml.h	use iwyu clean include (#27267)	5 years ago
nccl.cc	bug fix, test=develop (#28674)	5 years ago
nccl.h	add ncclSend and ncclRecv (#27621)	5 years ago
nvrtc.cc	remove WITH_DSO compile option (#25444)	5 years ago
nvrtc.h	use iwyu clean include (#27267)	5 years ago
tensorrt.cc	change avg pooling and global pooling to trt layer in dynamic shape mode (#28702)	5 years ago
tensorrt.h	TensorRT中ernie模型推理性能优化，支持变长输入 (#28367)	5 years ago
warpctc.cc	Fix the grammar in copyright. (#8403)	7 years ago
warpctc.h	add support to float64 input of warpctc op. (#27399)	5 years ago