parent
							
								
									3493f5bd24
								
							
						
					
					
						commit
						b16c0a84e6
					
				@ -1,46 +0,0 @@
 | 
				
			||||
/* Copyright (c) 2016 PaddlePaddle Authors. All Rights Reserve.
 | 
				
			||||
 | 
				
			||||
Licensed under the Apache License, Version 2.0 (the "License");
 | 
				
			||||
you may not use this file except in compliance with the License.
 | 
				
			||||
You may obtain a copy of the License at
 | 
				
			||||
 | 
				
			||||
    http://www.apache.org/licenses/LICENSE-2.0
 | 
				
			||||
 | 
				
			||||
Unless required by applicable law or agreed to in writing, software
 | 
				
			||||
distributed under the License is distributed on an "AS IS" BASIS,
 | 
				
			||||
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 | 
				
			||||
See the License for the specific language governing permissions and
 | 
				
			||||
limitations under the License. */
 | 
				
			||||
 | 
				
			||||
#include "CudnnConvBaseLayer.h"
 | 
				
			||||
 | 
				
			||||
namespace paddle {
 | 
				
			||||
/**
 | 
				
			||||
 * @brief A 2-dimension conv layer implemented by cuDNN. It only
 | 
				
			||||
 *        supports GPU mode. We automatic select CudnnConvLayer for GPU
 | 
				
			||||
 *        mode and ExpandConvLayer for CPU mode if you set type of "conv".
 | 
				
			||||
 *        User also can specfiy type of "exconv" or "cudnn_conv" for
 | 
				
			||||
 *        particular type.
 | 
				
			||||
 *
 | 
				
			||||
 * The config file api is img_conv_layer.
 | 
				
			||||
 */
 | 
				
			||||
class CudnnConvLayer : public CudnnConvBaseLayer {
 | 
				
			||||
public:
 | 
				
			||||
  explicit CudnnConvLayer(const LayerConfig& config)
 | 
				
			||||
      : CudnnConvBaseLayer(config) {}
 | 
				
			||||
 | 
				
			||||
  ~CudnnConvLayer() {}
 | 
				
			||||
};
 | 
				
			||||
 | 
				
			||||
REGISTER_LAYER(cudnn_conv, CudnnConvLayer);
 | 
				
			||||
 | 
				
			||||
class CudnnConvTransLayer : public CudnnConvBaseLayer {
 | 
				
			||||
public:
 | 
				
			||||
  explicit CudnnConvTransLayer(const LayerConfig& config)
 | 
				
			||||
      : CudnnConvBaseLayer(config) {}
 | 
				
			||||
 | 
				
			||||
  ~CudnnConvTransLayer() {}
 | 
				
			||||
};
 | 
				
			||||
 | 
				
			||||
REGISTER_LAYER(cudnn_convt, CudnnConvTransLayer);
 | 
				
			||||
}  // namespace paddle
 | 
				
			||||
					Loading…
					
					
				
		Reference in new issue