Method TestGradient

src/caffe/test/test_reduction_layer.cpp:80–92

  }

  void TestGradient(ReductionParameter_ReductionOp op,
                    float coeff = 1, int axis = 0) {
    typedef typename TypeParam::Dtype Dtype;
    LayerParameter layer_param;
    ReductionParameter* reduction_param = layer_param.mutable_reduction_param();
    reduction_param->set_operation(op);
    reduction_param->set_coeff(coeff);
    reduction_param->set_axis(axis);
    ReductionLayer<Dtype> layer(layer_param);
    GradientChecker<Dtype> checker(1e-2, 2e-3);
    checker.CheckGradientExhaustive(&layer, this->blob_bottom_vec_,
        this->blob_top_vec_);
  }

  Blob<Dtype>* const blob_bottom_;
  Blob<Dtype>* const blob_top_;