TOOL ERROR: cannot reshape array

[j0shm]

I have a TFLite that I can successfully analyze (targeting NUCLEO-L4R5ZI-P board) with X-CUBE-AI. I then quantize the TFLite model using the "full integer quantization" method from the docs. However, when I analyze this quantized model, I'm met with "TOOL ERROR: cannot reshape array". I validate using random numbers. I don't understand the problem as the quantized version runs absolutely fine on my laptop (with the exact same input + dataloader as the un-quantized version).

I'd be really grateful for any help with this - thanks!