Library/ManagedCuda/CudaDNN/CudaDNNNativeMethods.cs
/// <summary>
/// This routine computes the gradient of the softmax function.
/// </summary>
/// <param name="handle">Handle to a previously created cuDNN context.</param>
/// <param name="algorithm">Enumerant to specify the softmax algorithm.</param>
/// <param name="mode">Enumerant to specify the softmax mode.</param>
/// <param name="alpha">Pointer to scaling factors (in host memory) used to blend the computation
/// result with prior value in the output layer as follows: dstValue =
/// alpha[0]*result + beta[0]*priorDstValue. Please refer to this section for
/// additional details.</param>
/// <param name="yDesc">Handle to the previously initialized input tensor descriptor.</param>
/// <param name="y">Data pointer to GPU memory associated with the tensor descriptor srcDesc.</param>
/// <param name="dyDesc">Handle to the previously initialized input differential tensor descriptor.</param>
/// <param name="dy">Data pointer to GPU memory associated with the tensor descriptor srcDiffData.</param>
/// <param name="beta">Pointer to scaling factors (in host memory) used to blend the computation
/// result with prior value in the output layer as follows: dstValue =
/// alpha[0]*result + beta[0]*priorDstValue. Please refer to this section for
/// additional details.</param>
/// <param name="dxDesc">Handle to the previously initialized output differential tensor descriptor.</param>
/// <param name="dx">Data pointer to GPU memory associated with the output tensor descriptor destDiffDesc.</param>
This routine computes the gradient of the softmax function.
This routine computes the gradient of the softmax function .
[DllImport(CUDNN_API_DLL_NAME)]
		public static extern cudnnStatus cudnnSoftmaxBackward( cudnnHandle handle,
														cudnnSoftmaxAlgorithm algorithm,
														cudnnSoftmaxMode mode,
														ref double alpha,
														cudnnTensorDescriptor yDesc,
														CUdeviceptr y,
														cudnnTensorDescriptor dyDesc,
														CUdeviceptr dy,
														ref double beta,
														cudnnTensorDescriptor dxDesc,
														CUdeviceptr dx
													  );
[ DllImport ( CUDNN_API_DLL_NAME ) ] public static extern cudnnStatus cudnnSoftmaxBackward ( cudnnHandle handle , cudnnSoftmaxAlgorithm algorithm , cudnnSoftmaxMode mode , ref double alpha , cudnnTensorDescriptor yDesc , CUdeviceptr y , cudnnTensorDescriptor dyDesc , CUdeviceptr dy , ref double beta , cudnnTensorDescriptor dxDesc , CUdeviceptr dx ) ;
This routine computes the gradient of the softmax function.
This routine computes the gradient of the softmax function .
handle
cudnnHandle
Handle to a previously created cuDNN context.
Handle to a previously created cuDNN context .
algorithm
cudnnSoftmaxAlgorithm
Enumerant to specify the softmax algorithm.
Enumerant to specify the softmax algorithm .
mode
cudnnSoftmaxMode
Enumerant to specify the softmax mode.
Enumerant to specify the softmax mode .
double
ref
yDesc
cudnnTensorDescriptor
Handle to the previously initialized input tensor descriptor.
Handle to the previously initialized input tensor descriptor .
y
CUdeviceptr
Data pointer to GPU memory associated with the tensor descriptor srcDesc.
Data pointer to GPU memory associated with the tensor descriptor srcDesc .
dyDesc
cudnnTensorDescriptor
Handle to the previously initialized input differential tensor descriptor.
Handle to the previously initialized input differential tensor descriptor .
dy
CUdeviceptr
Data pointer to GPU memory associated with the tensor descriptor srcDiffData.
Data pointer to GPU memory associated with the tensor descriptor srcDiffData .
dxDesc
cudnnTensorDescriptor
Handle to the previously initialized output differential tensor descriptor.
Handle to the previously initialized output differential tensor descriptor .
dx
CUdeviceptr
Data pointer to GPU memory associated with the output tensor descriptor destDiffDesc.
Data pointer to GPU memory associated with the output tensor descriptor destDiffDesc .
alpha
Pointer to scaling factors (in host memory) used to blend the computation
result with prior value in the output layer as follows: dstValue
alpha[0]*result + beta[0]*priorDstValue. Please refer to this section for
additional details.
Pointer to scaling factors ( in host memory ) used to blend the computation result with prior value in the output layer as follows : dstValue alpha [ 0 ] * result + beta [ 0 ] * priorDstValue . Please refer to this section for additional details .
beta
Pointer to scaling factors (in host memory) used to blend the computation
result with prior value in the output layer as follows: dstValue
alpha[0]*result + beta[0]*priorDstValue. Please refer to this section for
additional details.
Pointer to scaling factors ( in host memory ) used to blend the computation result with prior value in the output layer as follows : dstValue alpha [ 0 ] * result + beta [ 0 ] * priorDstValue . Please refer to this section for additional details .