Acceleration platform is a custom processor. I am currently using the pytorch model, converting it to onnx/keras and then porting it to the processor for inference. But face this challenge of NHWC vs NHWC.
If possible,  can you please shed some light on the possibility of updating the model definition after permuting the layers, as mentioned in my original comment?
thanks!