How to prevent TFLite Converter from converting Conv1D to Conv2D and expand_dims

234 views

Skip to first unread message

John Reed

unread,

Oct 5, 2021, 8:47:01 AM10/5/21

to TensorFlow Lite

Hi all,

I was asked to implement support on TFLM for Conv1D op without using Conv2D. Since TFLite Converter converts Conv1D to Conv2D and expand_dims, is it possible to prevent the converter from doing this conversion so that I can try to implement Conv1D in TFLite and then port it to TFLM? Or are there other workarounds besides using Conv2D?