Caffe2 - C++ API
A deep learning, cross platform ML framework
Related Pages
Modules
Data Structures
Files
C++ API
Python API
GitHub
File List
Globals
torch
csrc
jit
variable_tensor_list.h
1
#pragma once
2
#include <ATen/ATen.h>
3
4
namespace
torch
{
5
namespace
jit {
6
7
// a wrapper to mark places where we expect all the at::Tensors to be
8
// variables
9
struct
variable_tensor_list
:
public
std::vector<at::Tensor> {
10
variable_tensor_list
() =
default
;
11
template
<
class
InputIt>
12
variable_tensor_list
(InputIt first, InputIt last)
13
: std::vector<at::Tensor>(first, last) {}
14
explicit
variable_tensor_list
(std::vector<at::Tensor>&& tensor)
15
: std::vector<at::Tensor>(std::move(tensor)) {}
16
};
17
18
}
// namespace jit
19
}
// namespace torch
torch::jit::variable_tensor_list
Definition:
variable_tensor_list.h:9
torch
Definition:
jit_type.h:17
Generated on Thu Mar 21 2019 13:06:24 for Caffe2 - C++ API by
1.8.11