Class LibMatrixCuDNNInputRowFetcher

  • All Implemented Interfaces:

    public class LibMatrixCuDNNInputRowFetcher
    extends LibMatrixCUDA
    implements AutoCloseable
    Performs a slice operation: out = in[(n+1):(n+1), 1:numColumns]
    • Constructor Detail

      • LibMatrixCuDNNInputRowFetcher

        public LibMatrixCuDNNInputRowFetcher​(GPUContext gCtx,
                                             String instName,
                                             MatrixObject image)
        Initialize the input fetcher
        gCtx - current gpu context
        instName - name of the instruction
        image - input matrix object.
    • Method Detail

      • getNthRow

        public jcuda.Pointer getNthRow​(int n)
        Copy the nth row and return the dense pointer
        n - zero-based row index
        dense pointer containing the nth row. This row is reused in the next iteration
      • close

        public void close()
        Deallocates temporary pointer
        Specified by:
        close in interface AutoCloseable