pooling: add breaks in switch

Author: vchuravy

src/operator/pooling.cc

@@ -15,10 +15,13 @@ Operator *CreateOp<cpu>(PoolingParam param, int dtype) {
     switch (param.pool_type) {
       case pool_enum::kMaxPooling:
         op = new PoolingOp<cpu, mshadow::red::maximum, DType>(param);
+        break;
       case pool_enum::kAvgPooling:
         op = new PoolingOp<cpu, mshadow::red::sum, DType>(param);
+        break;
       case pool_enum::kSumPooling:
         op = new PoolingOp<cpu, mshadow::red::sum, DType>(param);
+        break;
       default:
         LOG(FATAL) << "unknown activation type";
     }

src/operator/pooling.cu

@@ -20,11 +20,14 @@ Operator *CreateOp<gpu>(PoolingParam param, int dtype) {
     switch (param.pool_type) {
       case pool_enum::kMaxPooling:
         op = new CuDNNPoolingOp<DType>(param);
+        break;
       case pool_enum::kAvgPooling:
         op = new CuDNNPoolingOp<DType>(param);
+        break;
       case pool_enum::kSumPooling:
         LOG(WARNING) << "Sum pooling is not supported by cudnn, MxNet sum pooling is applied.";
         op = new PoolingOp<gpu, mshadow::red::sum, DType>(param);
+        break;
       default:
         LOG(FATAL) << "unknown pooling type";
     }
@@ -34,10 +37,13 @@ Operator *CreateOp<gpu>(PoolingParam param, int dtype) {
     switch (param.pool_type) {
       case pool_enum::kMaxPooling:
         op = new PoolingOp<gpu, mshadow::red::maximum, DType>(param);
+        break;
       case pool_enum::kAvgPooling:
         op = new PoolingOp<gpu, mshadow::red::sum, DType>(param);
+        break;
       case pool_enum::kSumPooling:
         op = new PoolingOp<gpu, mshadow::red::sum, DType>(param);
+        break;
       default:
         LOG(FATAL) << "unknown pooling type";
     }