const LayerParameter& param) {
ConvolutionParameter_Engine engine = param.convolution_param().engine();
if (engine == ConvolutionParameter_Engine_DEFAULT) {
-#ifdef CAFFE_ENGINE
engine = ConvolutionParameter_Engine_CAFFE;
-#endif
}
if (engine == ConvolutionParameter_Engine_CAFFE) {
return new CaffeConvolutionLayer<Dtype>(param);
const LayerParameter& param) {
PoolingParameter_Engine engine = param.pooling_param().engine();
if (engine == PoolingParameter_Engine_DEFAULT) {
-#ifdef CAFFE_ENGINE
engine = PoolingParameter_Engine_CAFFE;
-#endif
}
if (engine == PoolingParameter_Engine_CAFFE) {
return new CaffePoolingLayer<Dtype>(param);
const LayerParameter& param) {
ReLUParameter_Engine engine = param.relu_param().engine();
if (engine == ReLUParameter_Engine_DEFAULT) {
-#ifdef CAFFE_ENGINE
engine = ReLUParameter_Engine_CAFFE;
-#endif
}
if (engine == ReLUParameter_Engine_CAFFE) {
return new CaffeReLULayer<Dtype>(param);
const LayerParameter& param) {
SigmoidParameter_Engine engine = param.sigmoid_param().engine();
if (engine == SigmoidParameter_Engine_DEFAULT) {
-#ifdef CAFFE_ENGINE
engine = SigmoidParameter_Engine_CAFFE;
-#endif
}
if (engine == SigmoidParameter_Engine_CAFFE) {
return new CaffeSigmoidLayer<Dtype>(param);
const LayerParameter& param) {
TanHParameter_Engine engine = param.tanh_param().engine();
if (engine == TanHParameter_Engine_DEFAULT) {
-#ifdef CAFFE_ENGINE
engine = TanHParameter_Engine_CAFFE;
-#endif
}
if (engine == TanHParameter_Engine_CAFFE) {
return new CaffeTanHLayer<Dtype>(param);
const LayerParameter& param) {
SoftmaxParameter_Engine engine = param.softmax_param().engine();
if (engine == SoftmaxParameter_Engine_DEFAULT) {
-#ifdef CAFFE_ENGINE
engine = SoftmaxParameter_Engine_CAFFE;
-#endif
}
if (engine == SoftmaxParameter_Engine_CAFFE) {
return new CaffeSoftmaxLayer<Dtype>(param);