Caffe2 - C++ API
A deep learning, cross platform ML framework
Public Member Functions
caffe2::SwishFunctor< Context > Struct Template Reference

Public Member Functions

template<typename T >
bool operator() (const int N, const T *X, T *Y, Context *context) const
 
template<>
bool operator() (const int N, const T *X, T *Y, CPUContext *) const
 

Detailed Description

template<class Context>
struct caffe2::SwishFunctor< Context >

Definition at line 10 of file swish_op.h.


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