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.