C++ (Cpp) cudnnDestroyPoolingDescriptor Examples

Example #1

File: cudnn_SpatialMaxPooling.c Project: Aerobota/thnets

THFloatTensor *cudnn_SpatialMaxPooling_updateOutput(struct module *module, THFloatTensor *input)
{
	int kW = module->SpatialMaxPooling.kW;
	int kH = module->SpatialMaxPooling.kH;
	int dW = module->SpatialMaxPooling.dW;
	int dH = module->SpatialMaxPooling.dH;
	int padW = module->SpatialMaxPooling.padW;
	int padH = module->SpatialMaxPooling.padH;

	THFloatTensor *output = module->output;
	cudnnTensorDescriptor_t dinput, doutput;
	cudnnPoolingDescriptor_t dpool;
	float one = 1, zero = 0;
	int sizes[4];

	errcheck(THcudnn_TensorDescriptor(&dinput, input));
	errcheck(cudnnCreatePoolingDescriptor(&dpool));
	errcheck(cudnnSetPooling2dDescriptor(dpool, CUDNN_POOLING_MAX, kH, kW, padH, padW, dH, dW));
	errcheck(cudnnGetPoolingNdForwardOutputDim(dpool, dinput, 4, sizes));
	THCudaTensor_resize4d(output, sizes[0], sizes[1], sizes[2], sizes[3]);
	errcheck(THcudnn_TensorDescriptor(&doutput, output));

	errcheck(cudnnPoolingForward(THcudnn_getHandle(), dpool, &one, dinput, THFloatTensor_data(input), &zero,
		doutput, THFloatTensor_data(output)));

	cudnnDestroyTensorDescriptor(dinput);
	cudnnDestroyTensorDescriptor(doutput);
	cudnnDestroyPoolingDescriptor(dpool);
	return output;
}

Example #2

File: cudnn_pooling_layer.cpp Project: arjunds/NVIDIA-Jetson-TX1

CuDNNPoolingLayer<Dtype,Mtype>::~CuDNNPoolingLayer() {
  // Check that handles have been setup before destroying.
  if (!handles_setup_) { return; }

  cudnnDestroyTensorDescriptor(bottom_desc_);
  cudnnDestroyTensorDescriptor(top_desc_);
  cudnnDestroyPoolingDescriptor(pooling_desc_);
}

Example #3

File: cudnn_pooling_layer-inl.hpp Project: 52191114/cxxnet

 virtual ~CuDNNPoolingLayer(void) {
   CUDA_CHECK(cudnnDestroyTensorDescriptor(in_desc_));
   CUDA_CHECK(cudnnDestroyTensorDescriptor(out_desc_));
   CUDA_CHECK(cudnnDestroyPoolingDescriptor(pooling_desc_));
   CUDA_CHECK(cudnnDestroy(handle_));
 }

Example #4

File: cudnn.cpp Project: bebee/cudarray

PoolBC01CuDNN<T>::~PoolBC01CuDNN() {
  CUDNN_CHECK(cudnnDestroyTensorDescriptor(imgs_desc));
  CUDNN_CHECK(cudnnDestroyTensorDescriptor(poolout_desc));
  CUDNN_CHECK(cudnnDestroyPoolingDescriptor(pool_desc));
}