It seems there is a bug in TFLite BatchToSpaceNd to Relay conversion. I am specifically looking at this model - https://tfhub.dev/sayakpaul/lite-model/deeplabv3-mobilenetv2-int8/1/default/1
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?