iree-org/iree
Auf GitHub ansehen[TFL] `tfl.lstm` should have a stateful decomposition to other TFLite operations
Open
#9.190 geöffnet am 23. Mai 2022
bug 🐞help wantedintegrations/tosa
Repository-Metriken
- Stars
- (3.787 Stars)
- PR-Merge-Metriken
- (PR-Metriken ausstehend)
Beschreibung
Some significant models are blocked by tfl.lstm which can be decomposed into already support tflite operations.
Adding the decomposition should unblock these models.
/tmp/test.tflite:0:0: note: see current operation: %95 = "tfl.lstm"(%arg0, %44, %42, %41, %40, %44, %39, %38, %37, %44, %44, %44, %44, %92, %91, %90, %36, %44, %89, %88, %44, %87, %86, %85) ({
}) {asymmetric_quantize_inputs = false, cell_clip = 1.000000e+01 : f32, fused_activation_function = "TANH", kernel_type = "FULL", proj_clip = 0.000000e+00 : f32} : (tensor<1x1280xf32>, none, tensor<2048x1280xi8>, tensor<2048x1280xi8>, tensor<2048x1280xi8>, none, tensor<2048x640xi8>, tensor<2048x640xi8>, tensor<2048x640xi8>, none, none, none, none, tensor<2048xf32>, tensor<2048xf32>, tensor<2048xf32>, tensor<640x2048xi8>, none, tensor<1x640xf32>, tensor<1x2048xf32>, none, tensor<2048xf32>, tensor<2048xf32>, tensor<2048xf32>) -> tensor<*xf32>