Caffe2 - C++ API
A deep learning, cross platform ML framework
Public Member Functions
torch::jit::ArgumentSpec Struct Reference

Public Member Functions

 ArgumentSpec (bool with_grad, at::ArrayRef< IValue > inputs, size_t num_flat_inputs)
 
void addInput (const IValue &input, size_t &offset, bool with_grad)
 
void combineHash (const ArgumentInfo &arg)
 
bool operator== (const ArgumentSpec &spec) const
 
bool operator!= (const ArgumentSpec &spec) const
 
size_t size () const
 
const ArgumentInfoat (size_t i) const
 
size_t hashCode () const
 
std::vector< TypePtr > getTypes (Graph &graph) const
 

Detailed Description

Definition at line 69 of file argument_spec.h.


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