Mikhail Zolotukhin b763666f9f [JIT] Subgraph utils: add an optional vmap argument to the API to allow retrieving value mappings. (#43235) ... Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/43235 This functionality is needed when we want to not lose track of nodes/values as we merge and unmerge them into other nodes. For instance, if we have a side data structure with some meta information about values or nodes, this new functionality would allow to keep that metadata up to date after merging and unmerging nodes. Differential Revision: D23202648 Test Plan: Imported from OSS Reviewed By: eellison Pulled By: ZolotukhinM fbshipit-source-id: 350d21a5d462454166f8a61b51d833551c49fcc9		2020-08-25 18:13:29 -07:00
..
api	remove empty override pretty_print (#43341)	2020-08-20 18:48:29 -07:00
common
dist_autograd	Release DistAutogradContainer context for each dist_autograd test case (#38711)	2020-06-18 20:58:55 -07:00
jit	[JIT] Subgraph utils: add an optional vmap argument to the API to allow retrieving value mappings. (#43235)	2020-08-25 18:13:29 -07:00
rpc	Add set_device_map to TensorPipeOptions to support GPU args (#42637)	2020-08-14 18:46:55 -07:00
tensorexpr	Normalize loops with non-zero start. (#43179)	2020-08-21 12:37:27 -07:00
__init__.py	remediation of S205607	2020-07-17 17:19:47 -07:00