org.nd4j.linalg.api.ops

Class BaseOpContext

java.lang.Object

org.nd4j.linalg.api.ops.BaseOpContext

All Implemented Interfaces:

OpContext

public abstract class BaseOpContext
extends Object
implements OpContext

Implementation of common methods for OpContext

Author:

[email protected]

Field Summary

Fields

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

Constructor Summary

Constructors

Constructor and Description
BaseOpContext()

Method Summary

Modifier and Type	Method and Description
List<INDArray>	getInputArrays() This method returns List of input arrays defined within this context
List<INDArray>	getOutputArrays() This method returns List of output arrays defined within this context
void	setBArguments(boolean... arguments) This method sets boolean arguments required for operation
void	setIArguments(long... arguments) This method sets integer arguments required for operation
void	setInputArray(int index, INDArray array) This method adds INDArray as input argument for future op call
void	setInputArrays(INDArray... arrays) This method sets provided arrays as input arrays
void	setInputArrays(List<INDArray> arrays) This method sets provided arrays as input arrays
void	setOutputArray(int index, INDArray array) This method adds INDArray as output for future op call
void	setOutputArrays(INDArray... arrays) This method sets provided arrays as output arrays
void	setOutputArrays(List<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 org.nd4j.linalg.api.ops.OpContext

contextPointer, getRngStates, markInplace, setRngStates

Field Detail

fastpath_in

protected Map<Integer,INDArray> fastpath_in

fastpath_out

protected Map<Integer,INDArray> fastpath_out

fastpath_d

protected List<Double> fastpath_d

fastpath_b

protected List<Boolean> fastpath_b

fastpath_i

protected List<Long> fastpath_i

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

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

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

setInputArray

public void setInputArray(int index,
                          @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:

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
                           INDArray array)

Description copied from interface: OpContext

This method adds INDArray as output for future op call

Specified by:

setOutputArray in interface OpContext

setInputArrays

public void setInputArrays(@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
                            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