BlackCat_Tensors
A GPU-supported autograd and linear algebra library, designed for neural network construction
All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros
Namespaces | Functions
maxpooling.h File Reference
#include "common.h"
Include dependency graph for maxpooling.h:
This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Namespaces

 bc
 The Evaluator determines if an expression needs to be greedily optimized.
 
 bc::caffe
 

Functions

template<class Dtype >
void bc::caffe::MaxPoolForward (bc::host_tag, const Dtype *img_data, const int num, const int channels, const int height, const int width, const int pool_h, const int pool_w, const int krnl_h, const int krnl_w, const int stride_h, const int stride_w, const int pad_h, const int pad_w, Dtype *out_data, int *mask)
 
template<typename Dtype >
void bc::caffe::MaxPoolBackward (bc::host_tag, const Dtype *top_diff, const int *mask, const int num, const int channels, const int height, const int width, const int pool_h, const int pool_w, const int krnl_h, const int krnl_w, const int stride_h, const int stride_w, const int pad_h, const int pad_w, Dtype *bottom_diff)