How to make model use optimized Convolution2DTransposeBias Layer

TensorFlow General Discussion
1

Hello,

some of the mediapipe models use optimized layers such as Convolution2DTransposeBias.
If I create a model myself, is there a way I can make the tflite converter to use this optimized layer?

Relevant github issue:
https://github.com/google/mediapipe/issues/245#issuecomment-555227438

Thanks in advance!

3

@Kevin_Bondzio,

Welcome to the Tensorflow Forum!

We can use custom operators to allow conversion of an unsupported Tensorflow operator in Tensorflow Lite. If the flatbuffer file(.tflite) needs to be edited we can make use of schema and flatc command which helps us to convert a .tflite model to an editable .json.

The edited .json can be converted back to .tflite model.

Thank you!