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: