tiny_dnn  1.0.0
A header only, dependency-free deep learning framework in C++11
Classes | Public Member Functions | List of all members
tiny_dnn::core::OpKernelContext Class Reference

Classes

struct  OpParams
 

Public Member Functions

 OpKernelContext (const std::vector< tensor_t * > &in_data, std::vector< tensor_t * > &out_data)
 
 OpKernelContext (const std::vector< tensor_t * > &in_data, const std::vector< tensor_t * > &out_data, std::vector< tensor_t * > &out_grad, std::vector< tensor_t * > &in_grad)
 
tensor_t & input (const int idx) const
 
tensor_t & output (const int idx) const
 
tensor_t & input_grad (const int idx) const
 
tensor_t & output_grad (const int idx) const
 
void setParams (Params *params)
 
Paramsparams () const
 
void setParallelize (const bool parallelize)
 
bool parallelize () const
 
void setDevice (Device *device)
 
Devicedevice () const
 
void setLayer (layer *layer)
 
layerLayer () const
 
backend_t engine () const
 
void setEngine (const backend_t engine)
 

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