xla
Lower the embedding op
#6495
Merged

Commits
  • Lower embedding op
    bhavya01 committed 2 years ago
  • Update embedding op for 2D indices
    bhavya01 committed 2 years ago
  • lint fix
    bhavya01 committed 2 years ago
  • fix embedding op implementation
    bhavya01 committed 2 years ago
  • Check that xla embedding implementation is used
    bhavya01 committed 2 years ago
Loading