OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-07 12:21:27 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity
10,713 Commits 8,304 Branches 150 Tags 1.5 GiB
ce2854c875
Commit Graph

2 Commits

Author SHA1 Message Date
li-roy
ce2854c875 Create safe and unsafe versions of sparse_coo_tensor (#6058) 
Fixes #5748.

Added an unsafe version so embedding isn't slowed.

* Create safe and unsafe versions of sparse_coo_tensor

* rename sparse_coo_tensor_unsafe to _sparse_coo_tensor_unsafe

* refactor

* make helper static inline

* add sparse size check test

* fix lint
 2018-04-16 14:42:57 -04:00
Adam Paszke
1061d7970d Move broadcast and broadcast_coalesced to C++ 2018-01-18 11:16:45 +01:00