projects / platform / upstream / pytorch.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: b792bfe) | patch
Support tracing GenericList (#15969)
authorJames Reed <jamesreed@fb.com>
Wed, 16 Jan 2019 01:29:48 +0000 (17:29 -0800)
committerFacebook Github Bot <facebook-github-bot@users.noreply.github.com>
Wed, 16 Jan 2019 01:32:48 +0000 (17:32 -0800)
commitdc4977ddf0dd07a12961199a530015614fd37489
treeea941f01386ed06f80715fd1b8258bf48a5e9cedtree | snapshot
parentb792bfec0e2ef73b7bc156cdc6e18ef51962209ecommit | diff
Support tracing GenericList (#15969)

Summary:
Treat GenericList similarly to tuples and TensorList: recursively unpack them and assignValueTrace accordingly. Also add interpreter support for ListUnpack on GenericList
Pull Request resolved: https://github.com/pytorch/pytorch/pull/15969

Differential Revision: D13665139

Pulled By: jamesr66a

fbshipit-source-id: cd8cb3dd7475f424e48a69d217f2eac529df9f6a
test/test_jit.py diff | blob | history
torch/csrc/jit/register_prim_ops.cpp diff | blob | history
torch/csrc/jit/tracer.cpp diff | blob | history
Domain: Machine Learning / ML Framework;