Fix tf shared embedding (#17730)

Commit

3 years ago

Fix tf shared embedding (#17730) * fix the naming * from pt in test for now * make style * slow test and removed from_pt