PetaVision  Alpha
PVCuda::CudaTransposePoolingDeliverKernel Class Reference
Inheritance diagram for PVCuda::CudaTransposePoolingDeliverKernel:

Public Member Functions

 CudaTransposePoolingDeliverKernel (CudaDevice *inDevice)
 
void setArgs (PVLayerLoc const *preLoc, PVLayerLoc const *postLoc, PVLayerLoc const *origConnPreLoc, PVLayerLoc const *origConnPostLoc, int nxpPost, int nypPost, cudnnPoolingMode_t poolingMode, int multiplier, CudaBuffer *dataStoreBuffer, CudaBuffer *gSynBuffer, CudaBuffer *origConnDataStoreBuffer, CudaBuffer *origConnGSynBuffer, int channel)
 

Protected Member Functions

int calcBorderExcess (int preRestricted, int postRestricted, int border, int patchSizePostPerspective)
 
int calcManyScale (int preRestricted, int postRestricted)
 
int calcStride (int preRestricted, int postRestricted)
 
virtual int do_run () override
 

Protected Attributes

int mBorderExcessX = 0
 
int mBorderExcessY = 0
 
CudaBuffer * mCudnnDataStore = nullptr
 
CudaBuffer * mCudnnGSyn = nullptr
 
CudaBuffer * mCudnnOrigConnDataStore = nullptr
 
CudaBuffer * mCudnnOrigConnGSyn = nullptr
 
float * mDataStore = nullptr
 
cudnnTensorDescriptor_t mDataStoreDescriptor = nullptr
 
float * mGSyn = nullptr
 
cudnnTensorDescriptor_t mGSynDescriptor = nullptr
 
float mMultiplier = 1.0f
 
int mOrigConnBorderExcessX = 0
 
int mOrigConnBorderExcessY = 0
 
float * mOrigConnDataStore = nullptr
 
cudnnTensorDescriptor_t mOrigConnDataStoreDescriptor = nullptr
 
float * mOrigConnGSyn = nullptr
 
cudnnTensorDescriptor_t mOrigConnGSynDescriptor = nullptr
 
PVLayerLoc const * mOrigConnPostLoc = nullptr
 
PVLayerLoc const * mOrigConnPreLoc = nullptr
 
cudnnPoolingDescriptor_t mPoolingDescriptor = nullptr
 
cudnnPoolingMode_t mPoolingMode = CUDNN_POOLING_MAX
 
PVLayerLoc const * mPostLoc = nullptr
 
PVLayerLoc const * mPreLoc = nullptr
 

Detailed Description

Definition at line 15 of file CudaTransposePoolingDeliverKernel.hpp.


The documentation for this class was generated from the following files: