Skip to content

Pull requests: google-ai-edge/ai-edge-torch

New pull request New
44 Open 395 Closed
44 Open 395 Closed
Author
Filter by author
Loading
Label
Filter by label
Loading
Use alt + click/return to exclude labels
or ⇧ + click/return for logical OR
Projects
Filter by project
Loading
Milestones
Filter by milestone
Loading
Reviews
Filter by reviews
No reviews Review required Approved review Changes requested
Assignee
Filter by who’s assigned
Sort
Sort by
Newest Oldest Most commented Least commented Recently updated Least recently updated Best match
Most reactions
👍 👎 😄 🎉 😕 ❤️ 🚀 👀

Pull requests list

Add experimental API to odml torch export
#587 opened Apr 5, 2025 by copybara-service bot Loading…
Introduce sdpa_with_kv_update function. It will perform different cache update, pre/post sdpa logic based on kv cache layout.
#585 opened Apr 3, 2025 by copybara-service bot Loading…
Unify KV Cache classes between standard layers and experimental layers. Implement a single KVCache and KVCacheEntry class that can handle different layout (regular or transposed).
#581 opened Apr 2, 2025 by copybara-service bot Loading…
Update legalization rule for TFL op in StableHLO custom_call carrier
#568 opened Mar 20, 2025 by copybara-service bot Loading…
#odml-torch composite builder single boundary op
#559 opened Mar 18, 2025 by copybara-service bot Loading…
Add whisper encoder re-author with generative apis
#540 opened Mar 11, 2025 by bpkeene Loading…
1
Internal change for now
#536 opened Mar 7, 2025 by copybara-service bot Loading…
Internal only change
#535 opened Mar 6, 2025 by copybara-service bot Loading…
Fix for local attention sliding mask
#533 opened Mar 6, 2025 by copybara-service bot Loading…
Modified GemmaWrapper to pass local mask cache to original model.
#532 opened Mar 6, 2025 by copybara-service bot Loading…
Internal only change.
#513 opened Feb 21, 2025 by copybara-service bot Loading…
internal only change
#512 opened Feb 20, 2025 by copybara-service bot Loading…
Simplify TFLite model exporting.
#501 opened Feb 11, 2025 by copybara-service bot Loading…
Remove extra transpose ops
#498 opened Feb 7, 2025 by copybara-service bot Loading…
Add transformation to allow transposed input to BatchMatMul
#473 opened Jan 21, 2025 by copybara-service bot Loading…
Internal code test
#463 opened Jan 14, 2025 by copybara-service bot Loading…
Update README.md
#457 opened Jan 8, 2025 by dlgktjr Loading…
add experimental annotate_tensor api
#453 opened Jan 7, 2025 by copybara-service bot Loading…
Create SECURITY
#437 opened Dec 22, 2024 by Abirhasankas143 Loading…
2
Support passing attention mask as optional input in text_generator_main.cc.
#430 opened Dec 16, 2024 by copybara-service bot Loading…
Internal change. There is no other interesting things here.
#361 opened Nov 15, 2024 by copybara-service bot Loading…
Create a guitar to catch numerical regression for gemma2 (torch). Exclude the guitar test from being copybara-ed.
#352 opened Nov 12, 2024 by copybara-service bot Loading…
Update convert_to_tflite.py
#329 opened Oct 29, 2024 by haozha111 Loading…
internal change, not open sourced
#319 opened Oct 25, 2024 by copybara-service bot Loading…
Adds InferenceRunner class to execute the Stable Diffusion denoiser model via an InferenceCalculator instance.
#318 opened Oct 25, 2024 by copybara-service bot Loading…
ProTip! Add no:assignee to see everything that’s not assigned.