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

Public Member Functions

 MaxPoolFunctor (const OperatorBase &)
 
template<typename T , StorageOrder kOrder>
bool GlobalPoolingForward (int N, int C, int HxW, const T *X, T *Y, Context *context) const
 
template<typename T , StorageOrder kOrder>
bool Forward (int N, int C, const std::vector< int > &X_dims, const std::vector< int > &Y_dims, const std::vector< int > &kernel, const std::vector< int > &dilation, const std::vector< int > &stride, const std::vector< int > &pads, const T *X, T *Y, Context *context) const
 
template<typename T , StorageOrder kOrder>
bool GlobalPoolingBackward (int N, int C, int HxW, const T *dY, const T *X, const T *Y, T *dX, Context *context) const
 
template<typename T , StorageOrder kOrder>
bool Backward (int N, int C, const std::vector< int > &X_dims, const std::vector< int > &Y_dims, const std::vector< int > &kernel, const std::vector< int > &dilation, const std::vector< int > &stride, const std::vector< int > &pads, const T *dY, const T *X, const T *Y, T *dX, Context *context) const
 
template<>
bool GlobalPoolingBackward (const int N, const int C, const int HxW, const float *dY, const float *X, const float *Y, float *dX, CPUContext *) const
 
template<>
bool GlobalPoolingBackward (const int N, const int C, const int HxW, const float *dY, const float *X, const float *Y, float *dX, CPUContext *) const
 
template<>
bool Backward (const int N, const int C, const std::vector< int > &X_dims, const std::vector< int > &Y_dims, const std::vector< int > &kernel, const std::vector< int > &, const std::vector< int > &stride, const std::vector< int > &pads, const T *dY, const T *X, const T *Y, T *dX, CPUContext *) const
 
template<>
bool GlobalPoolingForward (const int N, const int C, const int HxW, const float *X, float *Y, CPUContext *context) const
 
template<>
bool GlobalPoolingForward (const int N, const int C, const int HxW, const float *X, float *Y, CPUContext *context) const
 

Detailed Description

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

Definition at line 259 of file pool_op.h.


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