pytorch/test/cpp
Raghavan Raman fa108bd264 Add flatten loops transformation (#46365)
Summary:
This diff removes the dependency of flattening on tensors by performing flattening on loops instead.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/46365

Reviewed By: ailzhang

Differential Revision: D24366347

Pulled By: navahgar

fbshipit-source-id: 4ba182f37212b6e4033cae13f8e75bc5144389f4
2020-10-16 17:05:26 -07:00
..
api [c++] Distance-agnostic triplet margin loss (#45377) 2020-09-30 12:37:35 -07:00
common
dist_autograd Fix Windows build failure after DDP PR merged (#45335) 2020-09-25 12:37:50 -07:00
jit [JIT] Fix Dict bug in constant hashing (#45929) 2020-10-07 17:40:17 -07:00
rpc Remove lock from GraphTask::set_exception_without_signal. (#45867) 2020-10-05 20:02:29 -07:00
tensorexpr Add flatten loops transformation (#46365) 2020-10-16 17:05:26 -07:00
__init__.py remediation of S205607 2020-07-17 17:19:47 -07:00