Visualize TensorRT engine's graph


Is there a tool to visualize a TensorRT engine graph? I’m used to visualizing models with Netron, but it doesn’t support TensorRT. Is there a way Nvidia supports visualizing TensorRT models?


TensorRT Version : 7.1.2
CUDA Version : 11.0
Operating System + Version : Ubuntu 18.04
Python Version (if applicable) : 3.6

Hi @weissrael,

We don’t have such a tool available for visualizing the TRT engine. In case you’re using tf-trt, you can visualize tf-trt optimized graph using tensorboard.

Thank you.