public class LibMatrixCuDNNInputRowFetcher extends LibMatrixCUDA implements AutoCloseable
cudaSupportFunctions, CUDNN_DATA_TYPE, customKernelSuffix, maxNumElementsOfCuDNNTensor, sizeOfDataType
Constructor and Description |
---|
LibMatrixCuDNNInputRowFetcher(GPUContext gCtx,
String instName,
org.apache.sysml.runtime.controlprogram.caching.MatrixObject image)
Initialize the input fetcher
|
Modifier and Type | Method and Description |
---|---|
void |
close()
Deallocates temporary pointer
|
jcuda.Pointer |
getNthRow(int n)
Copy the nth row and return the dense pointer
|
abs, acos, asin, atan, axpy, biasAdd, biasMultiply, cbind, ceil, channelSums, cos, cosh, dataTypePointerTo, double2float, exp, float2double, floor, getCublasHandle, getCudaKernels, getCusparseHandle, getDenseMatrixOutputForGPUInstruction, getDensePointer, getSparsePointer, isInSparseFormat, log, matmultTSMM, matrixMatrixArithmetic, matrixMatrixRelational, matrixScalarArithmetic, matrixScalarRelational, one, rbind, reluBackward, resetFloatingPointPrecision, round, sigmoid, sign, sin, sinh, sliceDenseDense, sliceOperations, sliceSparseDense, solve, sqrt, tan, tanh, toInt, transpose, unaryAggregate, zero
public LibMatrixCuDNNInputRowFetcher(GPUContext gCtx, String instName, org.apache.sysml.runtime.controlprogram.caching.MatrixObject image) throws DMLRuntimeException
gCtx
- current gpu contextinstName
- name of the instructionimage
- input matrix object.DMLRuntimeException
- if errorpublic jcuda.Pointer getNthRow(int n) throws DMLRuntimeException
n
- zero-based row indexDMLRuntimeException
- ?public void close()
close
in interface AutoCloseable
Copyright © 2017 The Apache Software Foundation. All rights reserved.