Calls updated to 4.0.5

1 files changed, 4 insertions, 3 deletions

diff --git a/ffi.lua b/ffi.lua
index c599e9d..22917e5 100644
--- a/ffi.lua
+++ b/ffi.lua
@@ -1183,9 +1183,10 @@ cudnnStatus_t             cudnnBatchNormalizationForwardInference(
 cudnnStatus_t             cudnnBatchNormalizationBackward(
                                 cudnnHandle_t                       handle,
                                 cudnnBatchNormMode_t                mode,
-
-                                const void                         *alpha,
-                                const void                         *beta,
+                                const void                         *alphaDataDiff,
+                                const void                         *betaDataDiff,
+                                const void                         *alphaParamDiff,
+                                const void                         *betaParamDiff,
 
                                 const cudnnTensorDescriptor_t       xDesc, /*  same desc for x, dx, dy */
                                 const void                         *x,