caffe2::AveragePoolFunctor< Context > Struct Template Reference

Public Member Functions
	AveragePoolFunctor (const OperatorBase &op)
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 *, const float *, float *dX, CPUContext *) const
template<>
bool	GlobalPoolingBackward (const int N, const int C, const int HxW, const float *dY, const float *, const float *, 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 *, const T *, 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

Data Fields
const bool	count_include_pad
Tensor	ones {Context::GetDeviceType()}

Detailed Description

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

Definition at line 199 of file pool_op.h.

---

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

• caffe2/operators/pool_op.h