BlackCat_Tensors
A GPU-supported autograd and linear algebra library, designed for neural network construction
|
#include "common.h"
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) |