public static extern CudnnStatus cudnnGetOutputTensor4dDim(
CudnnConvolutionDescriptorHandle convolutionDescriptor,
CudnnConvolutionPath path,
out int n,
out int c,
out int h,
out int w);
public CudnnConvolutionTensorDim( CudnnConvolutionPath path, int n, int c, int h, int w)
{
this.Path = path;
this.Num = n;
this.Channels = c;
this.Height = h;
this.Width = w;
}
public CudnnConvolutionTensorDim(CudnnConvolutionPath path, int n, int c, int h, int w)
{
this.Path = path;
this.Num = n;
this.Channels = c;
this.Height = h;
this.Width = w;
}
public CudnnConvolutionTensorDim GetOutputTensor(CudnnConvolutionPath path)
{
Contract.Ensures(Contract.Result<CudnnConvolutionTensorDim>() != null);
ThrowIfNotInitialized();
int n = 0, c = 0, h = 0, w = 0;
CudnnContext.Invoke(() => CudnnNativeMethods.cudnnGetOutputTensor4dDim(this.Handle, path, out n, out c, out h, out w));
return new CudnnConvolutionTensorDim(path, n, c, h, w);
}
public CudnnConvolutionTensorDim GetOutputTensor(CudnnConvolutionPath path)
{
Contract.Ensures(Contract.Result <CudnnConvolutionTensorDim>() != null);
ThrowIfNotInitialized();
int n = 0, c = 0, h = 0, w = 0;
CudnnContext.Invoke(() => CudnnNativeMethods.cudnnGetOutputTensor4dDim(this.Handle, path, out n, out c, out h, out w));
return(new CudnnConvolutionTensorDim(path, n, c, h, w));
}
public static extern CudnnStatus cudnnGetOutputTensor4dDim(
CudnnConvolutionDescriptorHandle convolutionDescriptor,
CudnnConvolutionPath path,
out int n,
out int c,
out int h,
out int w);
