My network contains some specific layers which are not supported by current tensorRT. so I want to run the conv layers and pooling layers on tensorRT and then use the output from tensorRT as the input of my caffe model which contains some specific layers. Is there some API or example code thar I can refer to? Thanks
Related topics
Topic | Replies | Views | Activity | |
---|---|---|---|---|
Caffe PReLU layer and tensorrt | 0 | 751 | May 26, 2017 | |
how to create custom layer in TensorRt by python | 0 | 583 | February 5, 2018 | |
Working with TensorFlow/UFF with custom layers | 0 | 858 | August 17, 2018 | |
What is a tensorcache file? | 0 | 683 | December 1, 2017 | |
How to parse the trained zf caffe model in tensorRT? | 4 | 1098 | January 6, 2020 | |
tensorRT doesn't support PReLu ? | 2 | 1475 | October 18, 2021 | |
a basic question about custom layers for Tensorflow UFF model | 4 | 631 | October 12, 2021 | |
TensorRT support for adding custom Tensorflow operations | 0 | 800 | September 28, 2017 | |
Can I reconstruct the network using tensorrt API when change a caffe mode to a tensorrt model? | 0 | 307 | November 9, 2020 | |
TensorRT with Tensorflow models - what are the options? | 1 | 607 | April 15, 2019 |