xla
b5c2a57a
- The XLA TensorImpl needs an empty Storage, as other APIs in PT wants to access it to fetch things like device or data type.
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Commit
7 years ago
The XLA TensorImpl needs an empty Storage, as other APIs in PT wants to access it to fetch things like device or data type. Also, a device guard is needed as Autograd Variable redirections want to create a device guard.
References
#185 - Prepare for XLA tensors
Author
dlibenzi
Committer
asuhan
Parents
caa3cc79
Loading