How to static assign specific operations and layers to DAL and GPU using TensorRT and PyTorch

Robotics & Edge Computing Jetson & Embedded Systems Jetson AGX Xavier
1

Hello, I am a student who is researching neural network acceleration.

When running AI models, I want to assign specific operations to NVDLA and GPU statically. However, accessing these settings in TensorRT can be difficult as they are often in compiled binaries. Can I get methods run using TensorRT, Pytorch, or Others

3

Hi,

You can find some examples in the below GitHub:

Thanks.

5

This topic was automatically closed 14 days after the last reply. New replies are no longer allowed.