Reshape operator is not supported by DLA

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?

Dear @Rhein,
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:
Package: nvidia-jetpack
Version: 5.0.2-b231
Architecture: arm64
Maintainer: NVIDIA Corporation
Installed-Size: 194
Depends: nvidia-jetpack-runtime (= 5.0.2-b231), nvidia-jetpack-dev (= 5.0.2-b231)
Priority: standard
Section: metapackages
Filename: pool/main/n/nvidia-jetpack/nvidia-jetpack_5.0.2-b231_arm64.deb
Size: 29304

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.

Dear @Rhein,
Yes. Please use shuffle layer.

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

Also check out the DLA github page for operator support : ONNX Operators on Orin DLA