|
| CudaPoolingDeliverKernel (CudaDevice *inDevice) |
|
void | setArgs (PVLayerLoc const *preLoc, PVLayerLoc const *postLoc, int nxpPost, int nypPost, cudnnPoolingMode_t poolingMode, int multiplier, CudaBuffer *inputBuffer, CudaBuffer *outputBuffer, int channel) |
|
|
static int | calcStride (int preRestricted, int postRestricted) |
|
|
int | calcBorderExcess (int preRestricted, int postRestricted, int border, int patchSizePostPerspective) |
|
int | calcManyScale (int preRestricted, int postRestricted) |
|
virtual int | do_run () override |
|
|
int | mBorderExcessX = 0 |
|
int | mBorderExcessY = 0 |
|
CudaBuffer * | mCudnnDataStore = nullptr |
|
CudaBuffer * | mCudnnGSyn = nullptr |
|
float * | mDataStore = nullptr |
|
cudnnTensorDescriptor_t | mDataStoreDescriptor = nullptr |
|
float * | mGSyn = nullptr |
|
cudnnTensorDescriptor_t | mGSynDescriptor = nullptr |
|
float | mMultiplier = 1.0f |
|
cudnnPoolingDescriptor_t | mPoolingDescriptor = nullptr |
|
cudnnPoolingMode_t | mPoolingMode = CUDNN_POOLING_MAX |
|
PVLayerLoc const * | mPostLoc = nullptr |
|
PVLayerLoc const * | mPreLoc = nullptr |
|
The documentation for this class was generated from the following files: