org.nd4j.linalg.jcublas.ops.executioner

Class JCudaExecutioner

java.lang.Object

org.nd4j.linalg.api.ops.executioner.DefaultOpExecutioner

org.nd4j.linalg.jcublas.ops.executioner.JCudaExecutioner

All Implemented Interfaces:

OpExecutioner

public class JCudaExecutioner
extends DefaultOpExecutioner

JCuda executioner.

Runs ops directly on the gpu

Author:

Adam Gibson

Nested Class Summary

Nested classes/interfaces inherited from interface org.nd4j.linalg.api.ops.executioner.OpExecutioner

OpExecutioner.ExecutionMode

Field Summary

Fields inherited from class org.nd4j.linalg.api.ops.executioner.DefaultOpExecutioner

executionMode, taskFactory

Constructor Summary

Constructors

Constructor and Description
JCudaExecutioner()

Method Summary

Modifier and Type	Method and Description
INDArray	exec(Accumulation op, int... dimension)
Op	exec(Op op)
Op	exec(Op op, int... dimension)
INDArray	execAndReturn(ScalarOp op, int... dimension)
INDArray	execAndReturn(TransformOp op)
INDArray	execAndReturn(TransformOp op, int... dimension)

Methods inherited from class org.nd4j.linalg.api.ops.executioner.DefaultOpExecutioner

exec, execAndReturn, execAndReturn, execAndReturn, execAndReturn, execAndReturn, executionMode, iterateOverAllColumns, iterateOverAllRows, parallelExecutioner, setExecutionMode

Methods inherited from class java.lang.Object

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

Constructor Detail

JCudaExecutioner

public JCudaExecutioner()

Method Detail

exec

public INDArray exec(Accumulation op,
                     int... dimension)

Specified by:

exec in interface OpExecutioner

Overrides:

exec in class DefaultOpExecutioner

execAndReturn

public INDArray execAndReturn(TransformOp op,
                              int... dimension)

Specified by:

execAndReturn in interface OpExecutioner

Overrides:

execAndReturn in class DefaultOpExecutioner

execAndReturn

public INDArray execAndReturn(ScalarOp op,
                              int... dimension)

Specified by:

execAndReturn in interface OpExecutioner

Overrides:

execAndReturn in class DefaultOpExecutioner

exec

public Op exec(Op op,
               int... dimension)

Specified by:

exec in interface OpExecutioner

Overrides:

exec in class DefaultOpExecutioner

exec

public Op exec(Op op)

Specified by:

exec in interface OpExecutioner

Overrides:

exec in class DefaultOpExecutioner

execAndReturn

public INDArray execAndReturn(TransformOp op)

Specified by:

execAndReturn in interface OpExecutioner

Overrides:

execAndReturn in class DefaultOpExecutioner