Function TEST

test/singa/test_operation_batchnorm.cc:30–46

#ifdef USE_DNNL
TEST(DNNLOperationBatchNorm, ForwardInference) {
  Tensor x(Shape{2, 2});
  Tensor alpha(Shape{2});
  Tensor beta(Shape{2});
  Tensor moving_mean(Shape{2});
  Tensor moving_var(Shape{2});

  Gaussian(0.0f, 1.0f, &x);
  Gaussian(0.0f, 1.0f, &alpha);
  Gaussian(0.0f, 1.0f, &beta);
  Gaussian(0.0f, 1.0f, &moving_mean);
  Gaussian(0.0f, 1.0f, &moving_var);

  BatchNormHandle batch_norm_handle(0u, x);
  Tensor y = CpuBatchNormForwardInference(batch_norm_handle, x, alpha, beta,
                                          moving_mean, moving_var);
}

TEST(DNNLOperationBatchNorm, ForwardTraining) {
  Tensor x(Shape{2, 2});