Raghavan Raman e795f88d6b [NNC] Make flatten transform in-place (#56629) ... Summary: Partial fix for https://github.com/pytorch/pytorch/issues/56157 This PR updates the `flatten` API in `LoopNest` to perform the flattening transformation in-place. After this transformation, the first loop in the input becomes the flattened loop. Pull Request resolved: https://github.com/pytorch/pytorch/pull/56629 Reviewed By: H-Huang Differential Revision: D28004787 Pulled By: navahgar fbshipit-source-id: 7474ae237fae3fff0cd1c64a276a8831dc5b7db0		2021-04-30 09:51:45 -07:00
..
api	Make PyTorch code-base clang-tidy compliant (#56892)	2021-04-28 14:10:25 -07:00
common
dist_autograd	Forbid trailing whitespace (#53406)	2021-03-05 17:22:55 -08:00
jit	[AutoAccept][Codemod][FBSourceClangFormatLinter] Daily arc lint --take CLANGFORMAT	2021-04-29 04:12:25 -07:00
lite_interpreter_runtime	Revert D27958477: [PyTorch][Edge] Add v4 and v5 models and remove unused model	2021-04-23 14:42:01 -07:00
rpc	[TensorPipe] Use targetDevice in tensorpipe_agent. (#56346)	2021-04-20 11:54:13 -07:00
tensorexpr	[NNC] Make flatten transform in-place (#56629)	2021-04-30 09:51:45 -07:00
__init__.py