It seems there is a bug in TFLite BatchToSpaceNd to Relay conversion. I am specifically looking at this model - Sayak | DeepLab v3 - Mobilenet v2 | Kaggle
The op where shape mismatches is this, first BatchToSpaceNd op
Relay output shape here is 1x66x66x384
@cchung100m CCing you as you might be interested.
Has anybody seen this error/interested in solving this?