search

google-ai-edge / ai-edge-torch

Supporting PyTorch models with the Google AI Edge TFLite runtime.
Apache License 2.0
310 stars 40 forks source link

Add fx pass to remove sdpa composite zero mask input #62

Closed chunnienc closed 3 months ago

chunnienc commented 3 months ago

BUG=https://b.corp.google.com/issues/347110617