xla
994f9fb6 - Filter tensor arguments from traced model. (#5689)

Commit
2 years ago
Filter tensor arguments from traced model. (#5689) * Filter tensor arguments from traced model. This PR filters tensor arguments from the list of arguments that would be given to the model. **Problem:** dynamo bridge assumed all arguments were tensors. **Solution:** filter tensor arguments so that we correctly collect tensor information. * Add test. * Fix lint issues. * Simplified test. * Use `openxla` instead of `openxla_eval` backend. * Rename variables for readability. * Use `openxla_eval` instead of `openxla`.
Author
Parents
Loading