xla
Add index to ATen XLA tensor
#350
Merged

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