diff options
| author | Adam Paszke <adam.paszke@gmail.com> | 2018-07-31 07:29:34 -0700 |
|---|---|---|
| committer | Facebook Github Bot <facebook-github-bot@users.noreply.github.com> | 2018-07-31 07:39:52 -0700 |
| commit | 5e5c15dd426ae608868ea76ec5be2d8817a31523 (patch) | |
| tree | 89e4fb2f2aaef7dc65e964ca15dbf05a833503c2 /tools/jit/templates | |
| parent | 6fb9acfc1654da214e2e45a72eda8dfe090bb466 (diff) | |
| download | pytorch-5e5c15dd426ae608868ea76ec5be2d8817a31523.tar.gz pytorch-5e5c15dd426ae608868ea76ec5be2d8817a31523.tar.bz2 pytorch-5e5c15dd426ae608868ea76ec5be2d8817a31523.zip | |
Add (constant size) TensorLists to JIT, use them in cat and stack nodes (#9948)
Summary:
zdevito
Pull Request resolved: https://github.com/pytorch/pytorch/pull/9948
Reviewed By: ezyang
Differential Revision: D9033666
Pulled By: apaszke
fbshipit-source-id: 02d75e391ed6dee62500842df50f0b6ee5e38846
Diffstat (limited to 'tools/jit/templates')
| -rw-r--r-- | tools/jit/templates/register_aten_ops.cpp | 7 |
1 files changed, 0 insertions, 7 deletions
diff --git a/tools/jit/templates/register_aten_ops.cpp b/tools/jit/templates/register_aten_ops.cpp index 06ad9c2840..2ff38e80b5 100644 --- a/tools/jit/templates/register_aten_ops.cpp +++ b/tools/jit/templates/register_aten_ops.cpp @@ -29,7 +29,6 @@ using autograd::Variable; using autograd::variable_list; using at::Scalar; using at::Tensor; -using at::TensorList; using at::TensorOptions; using at::DeviceGuard; @@ -42,12 +41,6 @@ int deviceForInputs(Stack & stack, size_t N) { return t.type().is_cuda() ? (int) t.get_device() : -1; } -std::vector<at::Tensor> toTensors(at::ArrayRef<IValue> ivalues) { - return fmap(ivalues, [](const IValue& v) { - return v.toTensor(); - }); -} - template<size_t N> std::array<bool, N> as_bool_array(const std::vector<int64_t>& vec) { std::array<bool, N> res; |