Caffe2 - C++ API
A deep learning, cross platform ML framework
Public Member Functions | Static Public Member Functions | Data Fields
torch::jit::python::IODescriptor::VariableMetadata Struct Reference

Public Member Functions

 VariableMetadata (const autograd::Variable &var)
 
bool operator== (const VariableMetadata &o) const
 

Static Public Member Functions

static size_t hash (const VariableMetadata &m)
 

Data Fields

std::vector< int64_t > sizes
 
at::ScalarType type
 
at::Device device
 
bool requires_grad
 

Detailed Description

Definition at line 17 of file python_arg_flatten.h.


The documentation for this struct was generated from the following file: