review.tizen.org Git - platform/upstream/pytorch.git/commit

projects / platform / upstream / pytorch.git / commit

author	Eric Faust <efaust@fb.com>
	Fri, 19 Apr 2019 06:48:59 +0000 (23:48 -0700)
committer	Facebook Github Bot <facebook-github-bot@users.noreply.github.com>
	Fri, 19 Apr 2019 06:52:00 +0000 (23:52 -0700)
commit	593bb145ceea548478f83fd820de44b95d375108
tree	5c88434f16733225cb3fcb3e873a55dd42a06472	tree \| snapshot
parent	9034b66f14096d39eb1a2bb8e273f4b78f93e58d	commit \| diff

Allow passing dicts as trace inputs. (#18092)

Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/18092

Previously, tracing required all inputs to be either tensors,
or tuples of tensor. Now, we allow users to pass dicts as well.

Differential Revision: D14491795

fbshipit-source-id: 7a2df218e5d00f898d01fa5b9669f9d674280be3

12 files changed:

Domain: Machine Learning / ML Framework;

RSS Atom

aten/src/ATen/core/type.cpp		diff \| blob \| history
test/cpp/jit/test_autodiff.h		diff \| blob \| history
test/test_jit.py		diff \| blob \| history
torch/autograd/function.py		diff \| blob \| history
torch/csrc/jit/pybind_utils.h		diff \| blob \| history
torch/csrc/jit/python_tracer.cpp		diff \| blob \| history
torch/csrc/jit/python_tracer.h		diff \| blob \| history
torch/csrc/jit/register_prim_ops.cpp		diff \| blob \| history
torch/csrc/jit/script/init.cpp		diff \| blob \| history
torch/csrc/jit/tracer.cpp		diff \| blob \| history
torch/csrc/jit/tracer.h		diff \| blob \| history
torch/jit/__init__.py		diff \| blob \| history