int dims = config.in_shape.size();
int spatial_dims = 2;
- batch_size_ = config.in_shape[0];
channels_ = config.channels;
pool_method_ = config.pool_method;
oclk_max_pool_forward.args(
count_,
ocl::KernelArg::PtrReadOnly(bottom),
- batch_size_,
channels_,
height_,
width_,
oclk_ave_pool_forward.args(
count_,
ocl::KernelArg::PtrReadOnly(bottom),
- batch_size_,
channels_,
height_,
width_,
oclk_sto_pool_forward.args(
count_,
ocl::KernelArg::PtrReadOnly(bottom),
- batch_size_,
channels_,
height_,
width_,
TEMPLATE(max_pool_forward, Dtype)
#endif
(
- const int nthreads, __global const Dtype* bottom_data, const int num,
+ const int nthreads, __global const Dtype* bottom_data,
const int channels, const int height, const int width,
const int pooled_height, const int pooled_width,
__global Dtype* top_data
#elif defined KERNEL_AVE_POOL
__kernel void TEMPLATE(ave_pool_forward, Dtype)(
- const int nthreads, __global const Dtype* const bottom_data, const int num,
+ const int nthreads, __global const Dtype* const bottom_data,
const int channels, const int height, const int width,
const int pooled_height, const int pooled_width,
__global Dtype* top_data)
#elif defined KERNEL_STO_POOL
__kernel void TEMPLATE(sto_pool_forward_test,Dtype)(
- const int nthreads, __global const Dtype* const bottom_data, const int num,
+ const int nthreads, __global const Dtype* const bottom_data,
const int channels, const int height, const int width,
const int pooled_height, const int pooled_width,
__global Dtype* top_data)