Add support for odml.update_external_kv_cache composite op in StableHLO to TFLite legalization.

google-ai-edge / ai-edge-torch

Supporting PyTorch models with the Google AI Edge TFLite runtime.

Apache License 2.0

378 stars 51 forks source link

Closed copybara-service[bot] closed 1 month ago

copybara-service[bot] commented 1 month ago

Add support for odml.update_external_kv_cache composite op in StableHLO to TFLite legalization.