tiny_dnn  1.0.0
A header only, dependency-free deep learning framework in C++11
Public Member Functions | Protected Attributes | List of all members
tiny_dnn::core::OpKernel Class Referenceabstract
Inheritance diagram for tiny_dnn::core::OpKernel:
Inheritance graph
[legend]
Collaboration diagram for tiny_dnn::core::OpKernel:
Collaboration graph
[legend]

Public Member Functions

 OpKernel (const OpKernelConstruction &context)
 
virtual void compute (const OpKernelContext &context)=0
 

Protected Attributes

Devicedevice_ = nullptr
 
Paramsparams_ = nullptr
 

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