Commit 83482a9
committed
Update on "[ET-VK] Add reshape functions for transformers related operators"
## Changes
* Implement resize functions for several operators used in Transformers models
## Motivation
Be able to support batched prefill for llama models.
Differential Revision: [D75686049](https://our.internmc.facebook.com/intern/diff/D75686049/)
[ghstack-poisoned]1 file changed
+1
-2
lines changed| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
1660 | 1660 | | |
1661 | 1661 | | |
1662 | 1662 | | |
1663 | | - | |
1664 | 1663 | | |
1665 | | - | |
| 1664 | + | |
1666 | 1665 | | |
1667 | 1666 | | |
1668 | 1667 | | |
| |||
0 commit comments