Caffe2 - C++ API
A deep learning, cross platform ML framework
Public Member Functions | Static Public Member Functions
torch::autograd::InputBuffer Struct Reference

Public Member Functions

 InputBuffer (size_t size)
 
 InputBuffer (const InputBuffer &other)=delete
 
 InputBuffer (InputBuffer &&other)=default
 
InputBufferoperator= (InputBuffer &&other)=default
 
void add (size_t pos, Variable var)
 
at::Device device () const
 
Variable operator[] (size_t pos)
 

Static Public Member Functions

static std::vector< Variablevariables (InputBuffer &&g)
 

Detailed Description

Definition at line 17 of file input_buffer.h.


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