xla
Add more ATen XLA comparison methods
#251
Merged

Add more ATen XLA comparison methods #251

asuhan merged 9 commits into master from asuhan/more_comparison
asuhan
asuhan Add more ATen XLA comparison methods
0508d562
asuhan Add ATen XLA comparison methods against scalar
2374b3f5
asuhan Share comparison op implementation with translator
c9772198
asuhan Use S32 as physical representation for comparison results
81c74bad
asuhan asuhan requested a review from dlibenzi dlibenzi 6 years ago
dlibenzi
dlibenzi requested changes on 2019-02-13
asuhan Only store logical element type instead of shape
ecf4074b
asuhan Add assertions in MakeTensorFromXlaLiteral
beecee80
asuhan Don't run shape inference in ComparisonOp
9389b586
asuhan Move logical_element_type to XLATensor::Data
6af04199
asuhan Make BuildComparisonOp more compact
53b858d0
dlibenzi
dlibenzi approved these changes on 2019-02-14
asuhan asuhan merged b07909a3 into master 6 years ago
asuhan asuhan deleted the asuhan/more_comparison branch 6 years ago

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
Milestone