I tried to convert a model to fully run on DLA, but error occured: “Layer ‘Reshape_351’ is not supported on DLA”. Can you suggest me any replacable operator for DLA?
Reshape operator is not supported by DLA
May I know Jetpack version? TRT 8.5 support Shuffle layer(onnx-tensorrt/builtin_op_importers.cpp at 7583da4c62475e84b7be31f4b8fb0c101873d434 · onnx/onnx-tensorrt · GitHub). Please check using Jetpack 5.1.
Thanks for your kind reply, SivaRamaKrishnaNV. Here are the details:
Maintainer: NVIDIA Corporation
Depends: nvidia-jetpack-runtime (= 5.0.2-b231), nvidia-jetpack-dev (= 5.0.2-b231)
Do you mean ‘reshape’ can be replaced by ‘shuffle’ which is supported by DLA with TRT 8.5?
Looking forward to your answer, thank you.
Yes. Please use shuffle layer.
This topic was automatically closed 14 days after the last reply. New replies are no longer allowed.