pytorch/test/cpp
Raghavan Raman 616afcf981 [jit] [shape analysis] Move constant tensors out of fused subgraphs during generalization (#70320)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/70320

ghstack-source-id: 146514368

Test Plan: `buck test mode/dev-nosan //caffe2/test/cpp/jit:jit`

Reviewed By: eellison

Differential Revision: D33280508

fbshipit-source-id: fe4291d7c49f0a498b330de96b698e99f6f6a505
2022-01-05 10:19:14 -08:00
..
api AdaptiveLogSoftmaxWithLoss no_batch_dim support (#69054) 2021-12-29 10:25:26 -08:00
c10d Make c10d tests -Werror clean (#69703) 2021-12-09 22:10:04 -08:00
common
dist_autograd Fix distributed autograd gradients synchronization (#57792) 2021-05-09 17:32:59 -07:00
jit [jit] [shape analysis] Move constant tensors out of fused subgraphs during generalization (#70320) 2022-01-05 10:19:14 -08:00
lazy [LTC] Upstream utils to extract BackendDevice from at::Tensor (#70069) 2021-12-23 12:42:03 -08:00
lite_interpreter_runtime Back out "Revert D30710710: [Pytorch Edge] Support profiling kineto events from external source" (#66421) 2021-10-12 10:55:29 -07:00
monitor Revert D33163841: Revert D33102715: Back out "Revert D32606547: torch/monitor: add C++ events and handlers" 2021-12-16 11:12:18 -08:00
rpc Remove ProcessGroup from TensorPipeAgent initialization (#68128) 2021-11-11 12:28:55 -08:00
tensorexpr [nnc] Add a run method to TensorExprKernel that takes in output tensors (#69477) 2021-12-22 00:30:15 -08:00
__init__.py