public abstract class BaseOpContext extends Object implements OpContext
Modifier and Type | Field and 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 and Description |
---|
BaseOpContext() |
Modifier and Type | Method and 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 |
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(INDArray... arrays)
This method sets provided arrays as input arrays
|
void |
setInputArrays(@NonNull List<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(INDArray... arrays)
This method sets provided arrays as output arrays
|
void |
setOutputArrays(@NonNull List<INDArray> arrays)
This method sets provided arrays as output arrays
|
void |
setTArguments(double... arguments)
This method sets floating point arguments required for operation
|
clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait
allowHelpers, contextPointer, getExecutionMode, getRngStates, markInplace, setExecutionMode, setRngStates, shapeFunctionOverride
close
protected ExecutionMode executionMode
public void setIArguments(long... arguments)
OpContext
setIArguments
in interface OpContext
public List<Long> getIArguments()
getIArguments
in interface OpContext
public int numIArguments()
numIArguments
in interface OpContext
public void setTArguments(double... arguments)
OpContext
setTArguments
in interface OpContext
public List<Double> getTArguments()
getTArguments
in interface OpContext
public int numTArguments()
numTArguments
in interface OpContext
public void setBArguments(boolean... arguments)
OpContext
setBArguments
in interface OpContext
public List<Boolean> getBArguments()
getBArguments
in interface OpContext
public int numBArguments()
numBArguments
in interface OpContext
public void setDArguments(DataType... arguments)
OpContext
setDArguments
in interface OpContext
public List<DataType> getDArguments()
getDArguments
in interface OpContext
public int numDArguments()
numDArguments
in interface OpContext
public void setInputArray(int index, @NonNull @NonNull INDArray array)
OpContext
setInputArray
in interface OpContext
public List<INDArray> getInputArrays()
OpContext
getInputArrays
in interface OpContext
public int numInputArguments()
numInputArguments
in interface OpContext
public INDArray getInputArray(int idx)
getInputArray
in interface OpContext
public List<INDArray> getOutputArrays()
OpContext
getOutputArrays
in interface OpContext
public void setOutputArray(int index, @NonNull @NonNull INDArray array)
OpContext
setOutputArray
in interface OpContext
public INDArray getOutputArray(int i)
getOutputArray
in interface OpContext
public int numOutputArguments()
numOutputArguments
in interface OpContext
public void setInputArrays(@NonNull @NonNull List<INDArray> arrays)
OpContext
setInputArrays
in interface OpContext
public void setOutputArrays(@NonNull @NonNull List<INDArray> arrays)
OpContext
setOutputArrays
in interface OpContext
public void setInputArrays(INDArray... arrays)
OpContext
setInputArrays
in interface OpContext
public void setOutputArrays(INDArray... arrays)
OpContext
setOutputArrays
in interface OpContext
Copyright © 2021. All rights reserved.