Fix fx tests with inputs_embeds (#31862)

Commit

2 years ago

Fix fx tests with inputs_embeds (#31862) * fix tests * [test_all] check * address review comments