Upgrading TLT exported models to work with TensorRT 7.1.2

Should be both.
I’m still thinking the solution for your special case.