Hi
I'm making a segmentation model with PyTorch (onnx).
Also, I use SDK 8.1
I have one problem.
The model that I making must use reshape layer without shape restrictions.
but, I know that the reshape layer only supports [1,1,1,N].
so, I am having a hard time making a segmentation model.
How can use reshape layer without shape restrictions in TIDL?
ex) [1, 4, 32, 1792] -> [1, 128, 32, 56]