Compile+train Keras model without REDUCE_PROD operation?

I’m trying to build a model and convert it to use in TensorFlow Lite for Microcontrollers. I’m having an issue where every Keras model I generate contains a REDUCE_PROD operator (even a completely basic model consisting of a single Dense(1) layer). However, the TF Lite for Microcontrollers runtime doesn’t support the REDUCE_PROD operator and flags an error upon attempting to load the model.

Is there a way I can exclude this operator when generating a model? Am I missing something?


submitted by /u/maha9000
[visit reddit] [comments]

Leave a Reply

Your email address will not be published.