xla
f996701a - Update Python device API for SPMD (#5129)

Commit

2 years ago

Update Python device API for SPMD (#5129) * Make python Api to respect the virtual device when SPMD is enabled * fix typo

References

Author

JackCaoG

Parents