github-featured/nvidia--tensorrt
2
0
Fork 0
Code Issues Pull Requests Deployments Models Agent Notes Packages Projects Releases Wiki Activity
43 Commits 42 Branches 72 Tags
20.02
Commit Graph

14 Commits

Author SHA1 Message Date
Kevin Chen b44715656a Fix assertion and documetation for batchedNMSPlugin (#407) 
* Fix assertion and documetation for batchedNMSPlugin

Update the assertion to validate input dimensions to handle UFF.
Clarify the purpose of clipBoxes field in plugin documentation.

Signed-off-by: Rajeev Rao <rajeevrao@nvidia.com>

* Address review comments

Co-authored-by: Rajeev Rao <rajeevsrao@users.noreply.github.com>
 2020-03-02 16:27:43 -08:00
Kevin Chen c59f2bca0c Ensure input and ouput types are the same for InstanceNormalizationPlugin (#356) 2020-01-23 11:14:12 -08:00
liji-nv 090231a93c Correctly set namespace when doing clone in instanceNormalizationPlugin (#318) 
Signed-off-by: Jin Li <liji@nvidia.com>
 2020-01-08 18:33:34 -08:00
Kevin Chen 5e289e919e Add support for multiple-namespace instantiation of plugins (#316) 2020-01-06 11:47:04 -08:00
Rajeev Rao 713475808d Fix memory leaks in instanceNormalizationPlugin #296 
Signed-off-by: Rajeev Rao <rajeevrao@nvidia.com>
 2020-01-06 10:59:55 -08:00
Rajeev Rao 572d54f917 TensorRT Open Source Release v7.0 
Release Notes:
https://docs.nvidia.com/deeplearning/sdk/tensorrt-release-notes/tensorrt-7.html

Signed-off-by: Rajeev Rao <rajeevrao@nvidia.com>
 2019-12-18 15:18:32 -08:00
Kevin Chen 98ab844a6c Adding support for building BERT plugins with GPU_ARCHS specified (#255) 2019-12-05 15:48:18 -08:00
Kevin Chen fb305dea28 Fixing compilation issue in embLayerNormPlugin (#248) 2019-12-03 10:45:49 -08:00
Pranav Marathe 77af534c21 Moves BERT plugins into main plugin library, removes demo (will be relocated to DeepLearningExamples shortly) 
Signed-off-by: pranavm <pranavm@nvidia.com>
 2019-11-13 13:48:15 -08:00
Kevin Chen 639d11abcc TensorRT Open Source Release/6.0 2019-09-16 14:00:59 -07:00
Rajeev Rao 443e495798 Merge release/5.1 changes into master 
Signed-off-by: Rajeev Rao <rajeevrao@nvidia.com>
 2019-08-27 00:26:56 -07:00
Tom Peters 296ee92fd7 initialize some variables 
Signed-off-by: Tom Peters <thomas.d.peters@gmail.com>
 2019-08-05 17:47:26 -07:00
Rajeev Rao b92c848b7f Merge release/5.1 changes into master (#34) 
* Anuragd/trt custom plugin (#21)

* Added support for custom TRT plugin in trtexec
* Restructed as a wrapper API for different platforms
* Review comments incorporated

Signed-off-by: Anurag Dixit <a.dixit91@gmail.com>

* Added support for writing calibration file (#23)

Signed-off-by: Anurag Dixit <a.dixit91@gmail.com>

* BERT example using TensorRT C++ API (#32)

This example demonstrates how to write a TensorRT model with custom
plugins to implement the BERT encoder with SQuAD output layer.

Signed-off-by: Rajeev Rao <rajeevrao@nvidia.com>

* Resolve undefined symbols in NvInfer plugin library

Signed-off-by: Rajeev Rao <rajeevrao@nvidia.com>
 2019-07-12 15:22:47 -07:00
Rajeev Rao 0d36bbb297 TensorRT Open Source Release/5.1 2019-06-17 10:47:48 -07:00