tensorflow / models

Models and examples built with TensorFlow
Other
76.95k stars 45.79k forks source link

pQRNN - TF Lite custom operator pooling_op missing #10652

Open Diego999 opened 2 years ago

Diego999 commented 2 years ago

Prerequisites

Please answer the following question for yourself before submitting an issue.

1. The entire URL of the documentation with the issue

https://github.com/tensorflow/models/tree/master/research/seq_flow_lite/tflite_ops and https://github.com/tensorflow/models/blob/master/research/seq_flow_lite/layers/qrnn_layers.py#L46

2. Describe the issue

The current version of pQRNN cannot be executed with Tensorflow Lite because there is a custom operator missing; all the others are provided in seq_flow_lite/tflite_ops.

Missing files: pooling_op.h and pooling_op.cc (or equivalent)

ginz commented 12 months ago

AFAIU, you should use Register_QRNN_POOLING from tflite_qrnn_pooling.h