| Modifier and Type | Method and Description |
|---|---|
static int |
JCudnn.cudnnCreateSpatialTransformerDescriptor(cudnnSpatialTransformerDescriptor stDesc) |
static int |
JCudnn.cudnnDestroySpatialTransformerDescriptor(cudnnSpatialTransformerDescriptor stDesc) |
static int |
JCudnn.cudnnSetSpatialTransformerNdDescriptor(cudnnSpatialTransformerDescriptor stDesc,
int samplerType,
int dataType,
int nbDims,
int[] dimA) |
static int |
JCudnn.cudnnSpatialTfGridGeneratorBackward(cudnnHandle handle,
cudnnSpatialTransformerDescriptor stDesc,
jcuda.Pointer dgrid,
jcuda.Pointer dtheta) |
static int |
JCudnn.cudnnSpatialTfGridGeneratorForward(cudnnHandle handle,
cudnnSpatialTransformerDescriptor stDesc,
jcuda.Pointer theta,
jcuda.Pointer grid) |
static int |
JCudnn.cudnnSpatialTfSamplerBackward(cudnnHandle handle,
cudnnSpatialTransformerDescriptor stDesc,
jcuda.Pointer alpha,
cudnnTensorDescriptor xDesc,
jcuda.Pointer x,
jcuda.Pointer beta,
cudnnTensorDescriptor dxDesc,
jcuda.Pointer dx,
jcuda.Pointer alphaDgrid,
cudnnTensorDescriptor dyDesc,
jcuda.Pointer dy,
jcuda.Pointer grid,
jcuda.Pointer betaDgrid,
jcuda.Pointer dgrid) |
static int |
JCudnn.cudnnSpatialTfSamplerForward(cudnnHandle handle,
cudnnSpatialTransformerDescriptor stDesc,
jcuda.Pointer alpha,
cudnnTensorDescriptor xDesc,
jcuda.Pointer x,
jcuda.Pointer grid,
jcuda.Pointer beta,
cudnnTensorDescriptor yDesc,
jcuda.Pointer y) |
Copyright © 2017. All rights reserved.