Skip to content

Commit

Permalink
Improved functional tensor geom transforms to work on floatX dtype (p…
Browse files Browse the repository at this point in the history 
…ytorch#2661)

* Improved functional tensor geom transforms to work on floatX dtype
- Fixes pytorch#2600
- added tests
- refactored test_affine

* Removed float16/cpu case
  • Loading branch information
@vfdev-5 @bryant1410
vfdev-5 authored and bryant1410 committed Nov 22, 2020
1 parent 514f43b commit 7ea4b90
Show file tree
Hide file tree
Showing 2 changed files with 243 additions and 175 deletions.
Loading

0 comments on commit 7ea4b90

Please sign in to comment.