Function TEST

test/singa/test_opencl_pooling.cc:32–57

using singa::Shape;

TEST(OpenclPooling, Setup) {
  OpenclPooling pool;
  EXPECT_EQ("OpenclPooling", pool.layer_type());

  singa::LayerConf conf;
  singa::PoolingConf *poolconf = conf.mutable_pooling_conf();
  poolconf->set_pool(singa::PoolingConf_PoolMethod_MAX);
  poolconf->set_kernel_h(1);
  poolconf->set_kernel_w(2);
  poolconf->set_pad_h(1);
  poolconf->set_pad_w(0);
  poolconf->set_stride_h(2);
  poolconf->set_stride_w(1);
  pool.Setup(Shape{1, 3, 3}, conf);

  EXPECT_EQ(singa::PoolingConf_PoolMethod_MAX, pool.pool_method());
  EXPECT_EQ(1u, pool.kernel_h());
  EXPECT_EQ(2u, pool.kernel_w());
  EXPECT_EQ(1u, pool.pad_h());
  EXPECT_EQ(0u, pool.pad_w());
  EXPECT_EQ(2u, pool.stride_h());
  EXPECT_EQ(1u, pool.stride_w());
  EXPECT_EQ(1u, pool.channels());
  EXPECT_EQ(3u, pool.height());
  EXPECT_EQ(3u, pool.width());
}

TEST(OpenclPooling, Forward) {
  const size_t batchsize = 2, c = 1, h = 3, w = 3;