SIGN IN SIGN UP

[ET-VK] Change weight packing in embedding

Pull Request resolved: https://github.com/pytorch/executorch/pull/7063

The existing weight tensor for aten.embedding is created using a `tensor_like` from the output tensor, which defaults to channel packed.

However, the weight tensor is actually a 2D-tensor of `(num_embedding, dim_of_embedding)`. It is better in space to use either width or height packing.

This diff changes the implementation to use height-packing.
ghstack-source-id: 255439082

Differential Revision: [D66421366](https://our.internmc.facebook.com/intern/diff/D66421366/)

Co-authored-by: Justin Yip <yipjustin@fb.com>
H
Hansong committed
633057cc188ec8693f018c667680b1e58710e9ee
Parent: 2a292c3
Committed by GitHub <noreply@github.com> on 11/26/2024, 11:19:13 PM