PR #350 Add index to ATen XLA tensor

Add index to ATen XLA tensor #350

asuhan merged 8 commits into master from asuhan/index_op

asuhan requested a review from

dlibenzi 6 years ago

asuhan force pushed from dd8c8129 to 11a1dca1 6 years ago

asuhan force pushed from 11a1dca1 to 951f0c58 6 years ago

dlibenzi requested changes on 2019-02-26

dlibenzi commented on 2019-02-26

dlibenzi requested changes on 2019-02-26

dlibenzi approved these changes on 2019-02-26

asuhan commented on 2019-02-26

Add index to ATen XLA tensor

39f0515d

Fix behavior with null indices

18040b97

Handle boolean mask indices

d963d291

Wrap around negative indices in the index method

a6ec0ccf

Transfer indices to the device of base, if needed

253c9f40

Consolidate index implementation in one place

44ed8b93

Additional minor cleanups

2cfbf004

Set the logical type for Data from ATen constructor

9c811b42

asuhan force pushed from 62323448 to 9c811b42 6 years ago

dlibenzi approved these changes on 2019-02-27

asuhan merged 66d2b9b2 into master 6 years ago

asuhan deleted the asuhan/index_op branch 6 years ago

Reviewers

dlibenzi

Assignees

No one assigned

Labels

None yet

Milestone

No milestone