xla
c57370dc - Don't move full tensor to device in deferred_init (#4819)

Commit

2 years ago

Don't move full tensor to device in deferred_init (#4819)

References

#4819 - [FSDP] Don't move full tensor to device in deferred_init

Author

Zach Zheng

Parents

Loading