You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
This issue originally reported by @tagunil has been moved to this dedicated repository for ai-edge-torch to enhance issue tracking and prioritization. To ensure continuity, we have created this new issue on your behalf.
We appreciate your understanding and look forward to your continued involvement.
1. System information
2. Code
tf.where(condition, x, y)
3. Failure after conversion
The model cannot run on TFLite Micro, as it doesn't support the old SELECT op, only the new SELECT_V2 one.
5. (optional) Any other info / logs
The change was introduced 5 years ago by @abattery to improve compatibility with the older runtimes: tensorflow/tensorflow@061d2e0
But now it breaks compatibility with TFLite Micro, which only supports the newer op.
Here's the legalization in question: https://github.com/tensorflow/tensorflow/blob/master/tensorflow/compiler/mlir/lite/transforms/legalize_patterns.td#L302
UPD: I believe that #93 is basically the same.
The text was updated successfully, but these errors were encountered: