Function TEST

test/singa/test_batchnorm.cc:29–42

using namespace singa;

TEST(BatchNorm, Setup) {
  BatchNorm batchnorm;
  // EXPECT_EQ("BatchNorm", batchnorm.layer_type());

  singa::LayerConf conf;
  singa::BatchNormConf *batchnorm_conf = conf.mutable_batchnorm_conf();
  batchnorm_conf->set_factor(0.01);
  batchnorm.Setup(Shape{2, 4, 4}, conf);

  EXPECT_FLOAT_EQ(0.01f, batchnorm.factor());
  EXPECT_EQ(2u, batchnorm.channels());
  EXPECT_EQ(4u, batchnorm.height());
  EXPECT_EQ(4u, batchnorm.width());
}

TEST(BatchNorm, Forward) {
  BatchNorm batchnorm;