0 0 1 Python

Integrate axis mapping in embedding op (#5440)

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

Update embedding op to support axis mapped textures.

Reviewed By: jorgep31415

Differential Revision: D62897503

fbshipit-source-id: 800507fd15b0d80f79f5928717cca1e262011bb7

Nathanael See committed 1y ago

41c21a40bee57356ab776bdfcb2f5a2aa85a9166

Parent: 3e1a578

Committed by Facebook GitHub Bot <facebook-github-bot@users.noreply.github.com> on 9/18/2024, 2:25:22 AM