Examples with CudnnHandle com.simiacryptus.mindseye.lang.cudnn.CudnnHandle used on opensource projects

Example 1 with CudnnHandle

use of com.simiacryptus.mindseye.lang.cudnn.CudnnHandle in project MindsEye by SimiaCryptus.

the class ImgTileAssemblyLayer method copy.

/**
 * Copy.
 *
 * @param copyParams the copy params
 */
public void copy(final CopyParams copyParams) {
    CudnnHandle gpu = copyParams.gpu;
    gpu.initThread();
    assert CudaDevice.isThreadDeviceId(gpu.getDeviceId());
    @Nullable final CudaTensor inputBuffer = gpu.getTensor(copyParams.inObj[copyParams.inputIndex].getData(), precision, MemoryType.Device, false);
    copy(gpu, copyParams.length, copyParams.tileDimensions, inputBuffer, copyParams.outputDims, copyParams.outputBuffer, copyParams.positionX, copyParams.totalHeight);
    inputBuffer.freeRef();
}