/// <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);
}
/// <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);
/// <summary>
/// </summary>
public DropoutDescriptor(cudnnHandle handle, cudnnDropoutDescriptor desc)
{
_handle = handle;
_desc = desc;
}