|
public GetPooling2dForwardOutputDim ( |
||
| inputTensorDesc | Handle to the previously initialized input tensor descriptor. | |
| outN | int | Number of images in the output |
| outC | int | Number of channels in the output |
| outH | int | Height of images in the output |
| outW | int | Width of images in the output |
| return | void | |
public void GetPooling2dForwardOutputDim(TensorDescriptor inputTensorDesc,
ref int outN,
ref int outC,
ref int outH,
ref int outW)
{
res = CudaDNNNativeMethods.cudnnGetPooling2dForwardOutputDim(_desc, inputTensorDesc.Desc, ref outN, ref outC, ref outH, ref outW);
Debug.WriteLine(String.Format("{0:G}, {1}: {2}", DateTime.Now, "cudnnGetPooling2dForwardOutputDim", res));
if (res != cudnnStatus.Success) throw new CudaDNNException(res);
}