public class CudaExecutioner extends DefaultOpExecutioner
OpExecutioner.ExecutionMode
Modifier and Type | Field and Description |
---|---|
protected ThreadLocal<org.bytedeco.javacpp.PointerPointer> |
extraz |
protected static NativeOps |
nativeOps |
protected Properties |
properties |
protected static TADManager |
tadManager |
executionMode
Constructor and Description |
---|
CudaExecutioner() |
Modifier and Type | Method and Description |
---|---|
INDArray |
exec(Accumulation op,
int... dimension) |
void |
exec(Aggregate op) |
<T extends Aggregate> |
exec(Batch<T> batch) |
INDArray |
exec(BroadcastOp op,
int... dimension) |
INDArray |
exec(IndexAccumulation op,
int... dimension) |
void |
exec(List<Aggregate> batch) |
Op |
exec(Op op) |
Op |
exec(Op op,
int... dimension) |
INDArray |
exec(RandomOp op)
This method executes specified RandomOp using default RNG available via Nd4j.getRandom()
|
INDArray |
exec(RandomOp op,
Random rng) |
INDArray |
execAndReturn(TransformOp op) |
protected <T extends Aggregate> |
getBuffer(Batch<T> batch) |
Properties |
getEnvironmentInformation()
This method return set of key/value and key/key/value objects, describing current environment
|
NativeOps |
getNativeOps() |
protected CudaContext |
intercept(ScalarOp op,
int[] dimension) |
protected CudaContext |
invoke(Accumulation op,
int[] dimension) |
protected CudaContext |
invoke(BroadcastOp op) |
protected CudaContext |
invoke(IndexAccumulation op,
int[] dimension) |
protected CudaContext |
invoke(ScalarOp op) |
protected CudaContext |
invoke(TransformOp op) |
protected INDArray |
naiveExec(Accumulation op,
int... dimension) |
checkForCompression, exec, exec, exec, execAndReturn, execAndReturn, execAndReturn, execAndReturn, execAndReturn, execAndReturn, executionMode, iterateOverAllColumns, iterateOverAllRows, setExecutionMode
protected static NativeOps nativeOps
protected static TADManager tadManager
protected ThreadLocal<org.bytedeco.javacpp.PointerPointer> extraz
protected transient volatile Properties properties
public NativeOps getNativeOps()
public INDArray exec(BroadcastOp op, int... dimension)
exec
in interface OpExecutioner
exec
in class DefaultOpExecutioner
protected INDArray naiveExec(Accumulation op, int... dimension)
op
- dimension
- public INDArray exec(Accumulation op, int... dimension)
exec
in interface OpExecutioner
exec
in class DefaultOpExecutioner
public INDArray exec(IndexAccumulation op, int... dimension)
exec
in interface OpExecutioner
exec
in class DefaultOpExecutioner
public Op exec(Op op, int... dimension)
exec
in interface OpExecutioner
exec
in class DefaultOpExecutioner
public Op exec(Op op)
exec
in interface OpExecutioner
exec
in class DefaultOpExecutioner
public INDArray execAndReturn(TransformOp op)
execAndReturn
in interface OpExecutioner
execAndReturn
in class DefaultOpExecutioner
protected CudaContext invoke(BroadcastOp op)
protected CudaContext invoke(IndexAccumulation op, int[] dimension)
protected CudaContext invoke(Accumulation op, int[] dimension)
protected CudaContext intercept(ScalarOp op, int[] dimension)
protected CudaContext invoke(ScalarOp op)
protected CudaContext invoke(TransformOp op)
protected <T extends Aggregate> DataBuffer getBuffer(Batch<T> batch)
public <T extends Aggregate> void exec(Batch<T> batch)
exec
in interface OpExecutioner
exec
in class DefaultOpExecutioner
public void exec(List<Aggregate> batch)
exec
in interface OpExecutioner
exec
in class DefaultOpExecutioner
public void exec(Aggregate op)
exec
in interface OpExecutioner
exec
in class DefaultOpExecutioner
public INDArray exec(RandomOp op)
exec
in interface OpExecutioner
exec
in class DefaultOpExecutioner
op
- public INDArray exec(RandomOp op, Random rng)
exec
in interface OpExecutioner
exec
in class DefaultOpExecutioner
public Properties getEnvironmentInformation()
getEnvironmentInformation
in interface OpExecutioner
getEnvironmentInformation
in class DefaultOpExecutioner
Copyright © 2016. All Rights Reserved.