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 | |
|---|---|---|---|---|
| Can I write a Custom Layer Using Python ? | 1 | 487 | June 6, 2019 | |
| customer layer questions | 21 | 1232 | October 18, 2021 | |
| A simple example on the custom layer API (TensorRT 2.1)? | 28 | 6832 | October 18, 2021 | |
| Problem for adding a caffe custom layer in TensorRT | 0 | 702 | April 24, 2019 | |
| How to create an custom plugin in TensorRT7 | 1 | 674 | April 10, 2020 | |
| TensorRT Plugin and custom layers | 5 | 4461 | February 5, 2018 | |
| How to add custom layers for a pytorch network? | 0 | 376 | September 17, 2019 | |
| Is there a completed document which illustrates the usage of addPlugin() api? | 1 | 527 | July 29, 2019 | |
| how to create custom layer in TensorRt by python | 0 | 613 | February 5, 2018 | |
| Import pytorch model in TensorRT and add custom plugin layer | 7 | 5418 | May 13, 2019 |