There is no official doc that introduces the way to add a custom layer which can do CUDA operations directly. Whether should I write .cu files or something else?
Related topics
Topic | Replies | Views | Activity | |
---|---|---|---|---|
Is there a completed document which illustrates the usage of addPlugin() api? | 1 | 502 | July 29, 2019 | |
How to add custom layers for a pytorch network? | 0 | 351 | September 17, 2019 | |
Writing TensorRT plugins for UffParser | 2 | 672 | October 12, 2021 | |
Can I write a Custom Layer Using Python ? | 1 | 457 | June 6, 2019 | |
TensorRT Plugin and custom layers | 5 | 4365 | February 5, 2018 | |
Add custom tactics for TensorRT | 3 | 494 | October 12, 2021 | |
Distributing/using CUDA inside a plugin? | 1 | 3390 | October 22, 2009 | |
Creating a Pluging layer in TensorRT | 2 | 3434 | October 12, 2021 | |
How to implement a custom plugin equivalent to "tf.tile" in Python API? | 5 | 724 | May 4, 2023 | |
TensorRT 3: Using add_plugin() in Python to add Custom Layer | 3 | 1535 | October 15, 2018 |