1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
| #pragma once
| #include <ATen/ATen.h>
|
| namespace torch {
| namespace jit {
|
| // a wrapper to mark places where we expect all the at::Tensors to be
| // variables
| struct variable_tensor_list : public std::vector<at::Tensor> {
| variable_tensor_list() = default;
| template <class InputIt>
| variable_tensor_list(InputIt first, InputIt last)
| : std::vector<at::Tensor>(first, last) {}
| explicit variable_tensor_list(std::vector<at::Tensor>&& tensor)
| : std::vector<at::Tensor>(std::move(tensor)) {}
| };
|
| } // namespace jit
| } // namespace torch
|
|
