Function TEST

test/singa/test_cudnn_pooling.cc:28–53

using singa::CudnnPooling;
using singa::Shape;
TEST(CudnnPooling, Setup) {
  CudnnPooling pool;
  //  EXPECT_EQ("CudnnPooling", 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(CudnnPooling, Forward) {
  const size_t batchsize = 1, c = 1, h = 3, w = 3;