Commit 55b3621
committed
Update base for 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 parent a54e556 commit 55b3621
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