示例#1
0
 /// <summary>
 /// </summary>
 public DropoutDescriptor(CudaDNNContext context)
 {
     _handle = context.Handle;
     _desc = new cudnnDropoutDescriptor();
     res = CudaDNNNativeMethods.cudnnCreateDropoutDescriptor(ref _desc);
     Debug.WriteLine(String.Format("{0:G}, {1}: {2}", DateTime.Now, "cudnnCreateDropoutDescriptor", res));
     if (res != cudnnStatus.Success) throw new CudaDNNException(res);
 }
示例#2
0
 /// <summary>
 /// </summary>
 public DropoutDescriptor(CudaDNNContext context)
 {
     _handle = context.Handle;
     _desc   = new cudnnDropoutDescriptor();
     res     = CudaDNNNativeMethods.cudnnCreateDropoutDescriptor(ref _desc);
     Debug.WriteLine(String.Format("{0:G}, {1}: {2}", DateTime.Now, "cudnnCreateDropoutDescriptor", res));
     if (res != cudnnStatus.Success)
     {
         throw new CudaDNNException(res);
     }
 }
 public static extern cudnnStatus cudnnCreateDropoutDescriptor(ref cudnnDropoutDescriptor dropoutDesc);
 public static extern cudnnStatus cudnnSetRNNDescriptor(cudnnRNNDescriptor rnnDesc,
                                                 int hiddenSize,
                                                 int seqLength,
                                                 int numLayers,
                                                 cudnnDropoutDescriptor dropoutDesc, // Between layers, not between recurrent steps.
                                                 cudnnRNNInputMode inputMode,
                                                 cudnnDirectionMode direction,
                                                 cudnnRNNMode mode,
                                                 cudnnDataType dataType);
 public static extern cudnnStatus cudnnSetDropoutDescriptor(cudnnDropoutDescriptor dropoutDesc,
                                                     cudnnHandle handle,
                                                     float dropout,
                                                     CUdeviceptr states,
                                                     SizeT stateSizeInBytes,
                                                     ulong seed);
 public static extern cudnnStatus cudnnDropoutForward(cudnnHandle handle,
                                               cudnnDropoutDescriptor dropoutDesc,
                                               cudnnTensorDescriptor xdesc,
                                               CUdeviceptr x,
                                               cudnnTensorDescriptor ydesc,
                                               CUdeviceptr y,
                                               CUdeviceptr reserveSpace,
                                               SizeT reserveSpaceSizeInBytes);
 public static extern cudnnStatus cudnnDropoutBackward(cudnnHandle handle,
                                        cudnnDropoutDescriptor dropoutDesc,
                                        cudnnTensorDescriptor dydesc,
                                        CUdeviceptr dy,
                                        cudnnTensorDescriptor dxdesc,
                                        CUdeviceptr dx,
                                        CUdeviceptr reserveSpace,
                                        SizeT reserveSpaceSizeInBytes);
 public static extern cudnnStatus cudnnDestroyDropoutDescriptor(cudnnDropoutDescriptor dropoutDesc);
示例#9
0
 /// <summary>
 /// </summary>
 public DropoutDescriptor(cudnnHandle handle, cudnnDropoutDescriptor desc)
 {
     _handle = handle;
     _desc   = desc;
 }