PR #6495 Lower the embedding op

Lower the embedding op #6495

bhavya01 merged 5 commits into master from embedding

Lower embedding op

a4b864c7

Update embedding op for 2D indices

f2bcf714

lint fix

f1d4569d

fix embedding op implementation

3088bebf

bhavya01 requested a review from

wonjoo-wj 1 year ago

bhavya01 assigned

bhavya01 1 year ago

Check that xla embedding implementation is used

e7e2c3bd

wonjoo-wj approved these changes on 2024-02-12

bhavya01 merged 657b6925 into master 1 year ago

Reviewers

wonjoo-wj

Assignees

bhavya01

Labels

None yet

Milestone

No milestone