diff options
| author | Boris Fomitchev <bfomitchev@nvidia.com> | 2015-11-13 01:25:43 +0300 |
|---|---|---|
| committer | Boris Fomitchev <bfomitchev@nvidia.com> | 2015-11-13 01:25:43 +0300 |
| commit | 413634aa8e27d4daed18d03e56da20046c62ce66 (patch) | |
| tree | 497c41f3f912e64459d3d013f939c7178f65b1f5 /ffi.lua | |
| parent | 09b428e5896f62f700e24aa3393ebdac75982f30 (diff) | |
Natalia's fixed for BN. Added bntest.lua
Diffstat (limited to 'ffi.lua')
| -rw-r--r-- | ffi.lua | 8 |
1 files changed, 4 insertions, 4 deletions
@@ -1072,7 +1072,7 @@ cudnnStatus_t cudnnBatchNormalizationForwardTraining( const cudnnTensorDescriptor_t xDesc, const void *x, /* NxCxHxW */ - /* const cudnnTensorDescriptor_t yDesc, */ + const cudnnTensorDescriptor_t yDesc, void *y, /* NxCxHxW */ /* Same shared desc for all the 6 tensors below in the argument list. */ @@ -1139,7 +1139,7 @@ cudnnStatus_t cudnnBatchNormalizationForwardInference( const cudnnTensorDescriptor_t xDesc, const void *x, /* NxCxHxW */ - /* const cudnnTensorDescriptor_t yDesc, */ + const cudnnTensorDescriptor_t yDesc, void *y, /* NxCxHxW */ /* Same desc for all 4 tensors below */ @@ -1188,9 +1188,9 @@ cudnnStatus_t cudnnBatchNormalizationBackward( const cudnnTensorDescriptor_t xDesc, /* same desc for x, dx, dy */ const void *x, - /* const cudnnTensorDescriptor_t dyDesc, */ + const cudnnTensorDescriptor_t dyDesc, const void *dy, - /* const cudnnTensorDescriptor_t dxDesc, */ + const cudnnTensorDescriptor_t dxDesc, void *dx, /* this tensor desc is used for all the 4 tensors below */ |