org.apache.sysml.runtime.matrix.data

Class LibMatrixCuDNNRnnAlgorithm

java.lang.Object

org.apache.sysml.runtime.matrix.data.LibMatrixCuDNNRnnAlgorithm

All Implemented Interfaces:

AutoCloseable

public class LibMatrixCuDNNRnnAlgorithm
extends Object
implements AutoCloseable

Constructor Summary

Constructors

Constructor and Description
LibMatrixCuDNNRnnAlgorithm(org.apache.sysml.runtime.controlprogram.context.ExecutionContext ec, GPUContext gCtx, String instName, String rnnMode, int N, int T, int M, int D, boolean isTraining, jcuda.Pointer w)

Method Summary

Modifier and Type	Method and Description
void	close()

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Detail

LibMatrixCuDNNRnnAlgorithm

public LibMatrixCuDNNRnnAlgorithm(org.apache.sysml.runtime.controlprogram.context.ExecutionContext ec,
                                  GPUContext gCtx,
                                  String instName,
                                  String rnnMode,
                                  int N,
                                  int T,
                                  int M,
                                  int D,
                                  boolean isTraining,
                                  jcuda.Pointer w)
                           throws DMLRuntimeException

Throws:

DMLRuntimeException

Method Detail

close

public void close()

Specified by:

close in interface AutoCloseable