BlackCat_Tensors
A GPU-supported autograd and linear algebra library, designed for neural network construction
|
Functions | |
bool | is_a_ge_zero_and_a_lt_b (int a, int b) |
template<class Dtype > | |
void | 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 | 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) |
|
inline |
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 | ||
) |
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 | ||
) |