Skip to content

Commit 01a4c3d

Browse files
authored
Update README.md (#31)
BUG=none
1 parent ffc6b9c commit 01a4c3d

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

ai_edge_torch/generative/layers/README.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -43,4 +43,4 @@ Currently, the library provides the following configuration class for you to cus
4343

4444
## High-Level function boundary for performance
4545
We introduce High-Level Function Boundary (HLFB) as a way of annotating performance-critical pieces of the model (e.g. `scaled_dot_product_attention`, or `KVCache`). HLFB allows the converter to lower the annotated blocks to performant TFLite custom ops. Following is an example of applying HLFB to `SDPA`:
46-
https://github.com/google-ai-edge/ai-edge-torch/blob/853301630f2b2455bd2e2f73d8a47e1a1534c91c/ai_edge_torch/generative/layers/attention.py#L74-L122
46+
https://github.com/google-ai-edge/ai-edge-torch/blob/25c764ad21e6f1fda5600dfc27406ef0424c8c3a/ai_edge_torch/generative/layers/scaled_dot_product_attention.py#L69-L117

0 commit comments

Comments
 (0)