/// <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; }