C++ (Cpp) LayerParameter::bn_param Examples

Example #1

File: layer_factory.cpp Project: pby5/caffe-mt

shared_ptr<Layer<Dtype> > GetBNLayer(const LayerParameter& param) {
  BNParameter_Type BN_type = param.bn_param().bn_type();
  if (BN_type == BNParameter_Type_CHANNEL_WISE) {
    return shared_ptr<Layer<Dtype> >(new ChannlWiseBNLayer<Dtype>(param));
  } else if (BN_type == BNParameter_Type_ELEMENT_WISE) {
    return shared_ptr<Layer<Dtype> >(new EltWiseBNLayer<Dtype>(param));
  } else {
    LOG(FATAL) << "Layer " << param.name() << " has unknown type.";
  }
}