org.nd4j.linalg.api.ops.executioner

Class DefaultOpExecutioner

java.lang.Object

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

All Implemented Interfaces:

OpExecutioner

public class DefaultOpExecutioner
extends Object
implements OpExecutioner

Basic op executioner. Knows how to iterate over the buffers of each respective ndarray and apply transformations

Author:

Adam Gibson

Nested Class Summary

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

OpExecutioner.ExecutionMode

Field Summary

Fields

Modifier and Type	Field and Description
protected OpExecutioner.ExecutionMode	executionMode
protected TaskFactory	taskFactory

Constructor Summary

Constructors

Constructor and Description
DefaultOpExecutioner()

Method Summary

Modifier and Type	Method and Description
protected void	doAccumulationOp(Accumulation op)
protected void	doBroadcastOp(BroadcastOp op)
protected void	doIndexAccumulationOp(IndexAccumulation op)
protected void	doScalarOp(ScalarOp op)
protected void	doTransformOp(TransformOp op)
INDArray	exec(Accumulation op, int... dimension) Execute an accumulation along one or more dimensions
INDArray	exec(IndexAccumulation op, int... dimension) Execute an index accumulation along one or more dimensions
Op	exec(Op op) Execute the operation
Op	exec(Op op, int... dimension) Execute the operation along 1 or more dimensions
Accumulation	execAndReturn(Accumulation op) Execute and return the result from an accumulation
INDArray	execAndReturn(BroadcastOp op) Execute and return the result from a vector op
IndexAccumulation	execAndReturn(IndexAccumulation op) Execute and return the result from an index accumulation
INDArray	execAndReturn(Op op) Execute and return a result ndarray from the given op
INDArray	execAndReturn(ScalarOp op) Execute and return the result from a scalar op
INDArray	execAndReturn(ScalarOp op, int... dimension) Execute and return the result from an scalar op
INDArray	execAndReturn(TransformOp op) Execute a TransformOp and return the result
INDArray	execAndReturn(TransformOp op, int... dimension) Execute a transform operation along one or more dimensions
OpExecutioner.ExecutionMode	executionMode() Get the execution mode for this execuioner
void	iterateOverAllColumns(Op op) Iterate over every column of every slice
void	iterateOverAllRows(Op op) Iterate over every row of every slice
ParallelExecutioner	parallelExecutioner()
void	setExecutionMode(OpExecutioner.ExecutionMode executionMode) Set the execution mode

Methods inherited from class java.lang.Object

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

Field Detail

executionMode

protected OpExecutioner.ExecutionMode executionMode

taskFactory

protected TaskFactory taskFactory

Constructor Detail

DefaultOpExecutioner

public DefaultOpExecutioner()

Method Detail

parallelExecutioner

public ParallelExecutioner parallelExecutioner()

Specified by:

parallelExecutioner in interface OpExecutioner

exec

public Op exec(Op op)

Description copied from interface: OpExecutioner

Execute the operation

Specified by:

exec in interface OpExecutioner

Parameters:

op - the operation to execute

execAndReturn

public INDArray execAndReturn(Op op)

Description copied from interface: OpExecutioner

Execute and return a result ndarray from the given op

Specified by:

execAndReturn in interface OpExecutioner

Parameters:

op - the operation to execute

Returns:

the result from the operation

iterateOverAllRows

public void iterateOverAllRows(Op op)

Description copied from interface: OpExecutioner

Iterate over every row of every slice

Specified by:

iterateOverAllRows in interface OpExecutioner

Parameters:

op - the operation to apply

iterateOverAllColumns

public void iterateOverAllColumns(Op op)

Description copied from interface: OpExecutioner

Iterate over every column of every slice

Specified by:

iterateOverAllColumns in interface OpExecutioner

Parameters:

op - the operation to apply

execAndReturn

public INDArray execAndReturn(TransformOp op)

Description copied from interface: OpExecutioner

Execute a TransformOp and return the result

Specified by:

execAndReturn in interface OpExecutioner

Parameters:

op - the operation to execute

execAndReturn

public Accumulation execAndReturn(Accumulation op)

Description copied from interface: OpExecutioner

Execute and return the result from an accumulation

Specified by:

execAndReturn in interface OpExecutioner

Parameters:

op - the operation to execute

Returns:

the accumulated result

execAndReturn

public INDArray execAndReturn(ScalarOp op)

Description copied from interface: OpExecutioner

Execute and return the result from a scalar op

Specified by:

execAndReturn in interface OpExecutioner

Parameters:

op - the operation to execute

Returns:

the accumulated result

execAndReturn

public IndexAccumulation execAndReturn(IndexAccumulation op)

Description copied from interface: OpExecutioner

Execute and return the result from an index accumulation

Specified by:

execAndReturn in interface OpExecutioner

Parameters:

op - the index accumulation operation to execute

Returns:

the accumulated index

execAndReturn

public INDArray execAndReturn(BroadcastOp op)

Description copied from interface: OpExecutioner

Execute and return the result from a vector op

Specified by:

execAndReturn in interface OpExecutioner

exec

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

Description copied from interface: OpExecutioner

Execute the operation along 1 or more dimensions

Specified by:

exec in interface OpExecutioner

Parameters:

op - the operation to execute

exec

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

Description copied from interface: OpExecutioner

Execute an accumulation along one or more dimensions

Specified by:

exec in interface OpExecutioner

Parameters:

op - the accumulation

dimension - the dimension

Returns:

the accmulation op

exec

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

Description copied from interface: OpExecutioner

Execute an index accumulation along one or more dimensions

Specified by:

exec in interface OpExecutioner

Parameters:

op - the index accumulation operation

dimension - the dimension/s to execute along

Returns:

result

execAndReturn

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

Description copied from interface: OpExecutioner

Execute a transform operation along one or more dimensions

Specified by:

execAndReturn in interface OpExecutioner

Parameters:

op - the operation to execute

execAndReturn

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

Description copied from interface: OpExecutioner

Execute and return the result from an scalar op

Specified by:

execAndReturn in interface OpExecutioner

Parameters:

op - the operation to execute

Returns:

the accumulated result

executionMode

public OpExecutioner.ExecutionMode executionMode()

Description copied from interface: OpExecutioner

Get the execution mode for this execuioner

Specified by:

executionMode in interface OpExecutioner

Returns:

the execution mode for this executioner

setExecutionMode

public void setExecutionMode(OpExecutioner.ExecutionMode executionMode)

Description copied from interface: OpExecutioner

Set the execution mode

Specified by:

setExecutionMode in interface OpExecutioner

Parameters:

executionMode - the execution mode

doTransformOp

protected void doTransformOp(TransformOp op)

doAccumulationOp

protected void doAccumulationOp(Accumulation op)

doScalarOp

protected void doScalarOp(ScalarOp op)

doIndexAccumulationOp

protected void doIndexAccumulationOp(IndexAccumulation op)

doBroadcastOp

protected void doBroadcastOp(BroadcastOp op)