xla
Add index to ATen XLA tensor
#350
Merged
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Overview
Commits
8
Changes
View On
GitHub
Commits
Add index to ATen XLA tensor
asuhan
committed
6 years ago
Fix behavior with null indices
asuhan
committed
6 years ago
Handle boolean mask indices
asuhan
committed
6 years ago
Wrap around negative indices in the index method
asuhan
committed
6 years ago
Transfer indices to the device of base, if needed
asuhan
committed
6 years ago
Consolidate index implementation in one place
asuhan
committed
6 years ago
Additional minor cleanups
asuhan
committed
6 years ago
Set the logical type for Data from ATen constructor
asuhan
committed
6 years ago
Loading