Kevin Chen 98ab844a6c Adding support for building BERT plugins with GPU_ARCHS specified (#255)

2019-12-05 15:48:18 -08:00

..

kernels

Adding support for building BERT plugins with GPU_ARCHS specified (#255)

2019-12-05 15:48:18 -08:00

bboxUtils.h

TensorRT Open Source Release/5.1

2019-06-17 10:47:48 -07:00

bertCommon.h

Adding support for building BERT plugins with GPU_ARCHS specified (#255)

2019-12-05 15:48:18 -08:00

CMakeLists.txt

Adding support for building BERT plugins with GPU_ARCHS specified (#255)

2019-12-05 15:48:18 -08:00

cub_helper.h

TensorRT Open Source Release/5.1

2019-06-17 10:47:48 -07:00

cudaDriverWrapper.cu

TensorRT Open Source Release/6.0

2019-09-16 14:00:59 -07:00

cudaDriverWrapper.h

TensorRT Open Source Release/6.0

2019-09-16 14:00:59 -07:00

nmsHelper.cpp

TensorRT Open Source Release/5.1

2019-06-17 10:47:48 -07:00

nmsUtils.h

TensorRT Open Source Release/5.1

2019-06-17 10:47:48 -07:00

plugin.h

Moves BERT plugins into main plugin library, removes demo (will be relocated to DeepLearningExamples shortly)

2019-11-13 13:48:15 -08:00

serialize.hpp

TensorRT Open Source Release/6.0

2019-09-16 14:00:59 -07:00