Package org.nd4j.linalg.api.ops

Class BaseOpContext

java.lang.Object

org.nd4j.linalg.api.ops.BaseOpContext

All Implemented Interfaces:

AutoCloseable, OpContext

public abstract class BaseOpContext
extends Object
implements OpContext

Field Summary

Fields

Modifier and Type	Field	Description
protected ExecutionMode	executionMode
protected List<Boolean>	fastpath_b
protected List<DataType>	fastpath_d
protected List<Long>	fastpath_i
protected Map<Integer,INDArray>	fastpath_in
protected Map<Integer,INDArray>	fastpath_out
protected List<Double>	fastpath_t

Constructor Summary

Constructors

Constructor	Description
BaseOpContext()

Method Summary

Modifier and Type	Method	Description
List<Boolean>	getBArguments()
List<DataType>	getDArguments()
List<Long>	getIArguments()
INDArray	getInputArray(int idx)
List<INDArray>	getInputArrays()	This method returns List of input arrays defined within this context
INDArray	getOutputArray(int i)
List<INDArray>	getOutputArrays()	This method returns List of output arrays defined within this context
List<Double>	getTArguments()
int	numBArguments()
int	numDArguments()
int	numIArguments()
int	numInputArguments()
int	numOutputArguments()
int	numTArguments()
void	purge()	This method removes all in/out arrays from this OpContext
void	setArgs(INDArray[] inputArrs, long[] iArgs, DataType[] dArgs, double[] tArgs, boolean[] bArgs)	set context arguments
void	setBArguments(boolean... arguments)	This method sets boolean arguments required for operation
void	setDArguments(DataType... arguments)	This method sets data type arguments required for operation
void	setIArguments(long... arguments)	This method sets integer arguments required for operation
void	setInputArray(int index, @NonNull INDArray array)	This method adds INDArray as input argument for future op call
void	setInputArrays(@NonNull List<INDArray> arrays)	This method sets provided arrays as input arrays
void	setInputArrays(INDArray... arrays)	This method sets provided arrays as input arrays
void	setOutputArray(int index, @NonNull INDArray array)	This method adds INDArray as output for future op call
void	setOutputArrays(@NonNull List<INDArray> arrays)	This method sets provided arrays as output arrays
void	setOutputArrays(INDArray... arrays)	This method sets provided arrays as output arrays
void	setTArguments(double... arguments)	This method sets floating point arguments required for operation

Methods inherited from class java.lang.Object

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

Methods inherited from interface java.lang.AutoCloseable

close

Methods inherited from interface org.nd4j.linalg.api.ops.OpContext

allowHelpers, contextPointer, getExecutionMode, getRngStates, markInplace, setExecutionMode, setRngStates, shapeFunctionOverride

Field Detail

fastpath_in

protected Map<Integer,INDArray> fastpath_in

fastpath_out

protected Map<Integer,INDArray> fastpath_out

fastpath_t

protected List<Double> fastpath_t

fastpath_b

protected List<Boolean> fastpath_b

fastpath_i

protected List<Long> fastpath_i

fastpath_d

protected List<DataType> fastpath_d

executionMode

protected ExecutionMode executionMode

Constructor Detail

BaseOpContext

public BaseOpContext()

Method Detail

setIArguments

public void setIArguments(long... arguments)

Description copied from interface: OpContext

This method sets integer arguments required for operation

Specified by:

setIArguments in interface OpContext

getIArguments

public List<Long> getIArguments()

Specified by:

getIArguments in interface OpContext

numIArguments

public int numIArguments()

Specified by:

numIArguments in interface OpContext

setTArguments

public void setTArguments(double... arguments)

Description copied from interface: OpContext

This method sets floating point arguments required for operation

Specified by:

setTArguments in interface OpContext

getTArguments

public List<Double> getTArguments()

Specified by:

getTArguments in interface OpContext

numTArguments

public int numTArguments()

Specified by:

numTArguments in interface OpContext

setBArguments

public void setBArguments(boolean... arguments)

Description copied from interface: OpContext

This method sets boolean arguments required for operation

Specified by:

setBArguments in interface OpContext

getBArguments

public List<Boolean> getBArguments()

Specified by:

getBArguments in interface OpContext

numBArguments

public int numBArguments()

Specified by:

numBArguments in interface OpContext

setDArguments

public void setDArguments(DataType... arguments)

Description copied from interface: OpContext

This method sets data type arguments required for operation

Specified by:

setDArguments in interface OpContext

getDArguments

public List<DataType> getDArguments()

Specified by:

getDArguments in interface OpContext

numDArguments

public int numDArguments()

Specified by:

numDArguments in interface OpContext

setInputArray

public void setInputArray(int index,
                          @NonNull
                          @NonNull INDArray array)

Description copied from interface: OpContext

This method adds INDArray as input argument for future op call

Specified by:

setInputArray in interface OpContext

getInputArrays

public List<INDArray> getInputArrays()

Description copied from interface: OpContext

This method returns List of input arrays defined within this context

Specified by:

getInputArrays in interface OpContext

Returns:

numInputArguments

public int numInputArguments()

Specified by:

numInputArguments in interface OpContext

getInputArray

public INDArray getInputArray(int idx)

Specified by:

getInputArray in interface OpContext

getOutputArrays

public List<INDArray> getOutputArrays()

Description copied from interface: OpContext

This method returns List of output arrays defined within this context

Specified by:

getOutputArrays in interface OpContext

Returns:

setOutputArray

public void setOutputArray(int index,
                           @NonNull
                           @NonNull INDArray array)

Description copied from interface: OpContext

This method adds INDArray as output for future op call

Specified by:

setOutputArray in interface OpContext

getOutputArray

public INDArray getOutputArray(int i)

Specified by:

getOutputArray in interface OpContext

numOutputArguments

public int numOutputArguments()

Specified by:

numOutputArguments in interface OpContext

setInputArrays

public void setInputArrays(@NonNull
                           @NonNull List<INDArray> arrays)

Description copied from interface: OpContext

This method sets provided arrays as input arrays

Specified by:

setInputArrays in interface OpContext

setOutputArrays

public void setOutputArrays(@NonNull
                            @NonNull List<INDArray> arrays)

Description copied from interface: OpContext

This method sets provided arrays as output arrays

Specified by:

setOutputArrays in interface OpContext

setInputArrays

public void setInputArrays(INDArray... arrays)

Description copied from interface: OpContext

This method sets provided arrays as input arrays

Specified by:

setInputArrays in interface OpContext

setOutputArrays

public void setOutputArrays(INDArray... arrays)

Description copied from interface: OpContext

This method sets provided arrays as output arrays

Specified by:

setOutputArrays in interface OpContext

purge

public void purge()

Description copied from interface: OpContext

This method removes all in/out arrays from this OpContext

Specified by:

purge in interface OpContext

setArgs

public void setArgs(INDArray[] inputArrs,
                    long[] iArgs,
                    DataType[] dArgs,
                    double[] tArgs,
                    boolean[] bArgs)

Description copied from interface: OpContext

set context arguments

Specified by:

setArgs in interface OpContext