title: Converting pretrained saved model from NCHW to NHWC to make it compatible for Tensorflow Lite
body: <p>I have converted a model from PyTorch to Keras and used the backend to extract the tensorflow graph. Since the data format for PyTorch was NCHW, the model extracted and saved is also that. While converting the model to TFLite, due to the format being NCHW, it cannot get converted. Is there a way to convert the whole graph into NHCW?</p>