xla
Lower whole convolution to XLA
#955
Merged

Lower whole convolution to XLA #955

ailzhang merged 10 commits into pytorch:master from ailzhang:lower_convolution
ailzhang
init working lowering with one numerical diff in a test
6a8a71c0
fixes test; rename; format code
cc56a821
ailzhang ailzhang requested a review from asuhan asuhan 6 years ago
ailzhang ailzhang requested a review from dlibenzi dlibenzi 6 years ago
Merge branch 'master' of https://github.com/pytorch/xla into lower_co…
e3667a9e
dlibenzi
dlibenzi requested changes on 2019-08-23
fix transposed depthwise conv; remove extra couts
474f4765
address comments
0e41c1a6
ailzhang
dlibenzi
dlibenzi commented on 2019-08-25
dlibenzi
dlibenzi commented on 2019-08-25
dlibenzi
dlibenzi commented on 2019-08-25
dlibenzi
dlibenzi commented on 2019-08-25
dlibenzi
dlibenzi commented on 2019-08-25
address comments
e10dfb4a
remove extra comment
8102df85
ailzhang ailzhang force pushed from 8e6b138a to 8102df85 6 years ago
remove extra consts
c0bed5d4
dlibenzi
dlibenzi commented on 2019-08-27
use XLA_ERROR
947bcbc1
dlibenzi
dlibenzi approved these changes on 2019-08-27
another shape that mkldnn doesn't like
65e3d7d2
ailzhang
ailzhang ailzhang merged b986a467 into master 6 years ago
dlibenzi
dlibenzi

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
Milestone