public abstract class BaseOpContext extends Object implements OpContext
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 and Description |
---|
BaseOpContext() |
Modifier and Type | Method and Description |
---|---|
List<Boolean> |
getBArguments() |
List<Long> |
getIArguments() |
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
|
List<Double> |
getTArguments() |
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
|
clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait
contextPointer, getRngStates, markInplace, setRngStates
close
public void setIArguments(long... arguments)
OpContext
setIArguments
in interface OpContext
public List<Long> getIArguments()
getIArguments
in interface OpContext
public void setTArguments(double... arguments)
OpContext
setTArguments
in interface OpContext
public List<Double> getTArguments()
getTArguments
in interface OpContext
public void setBArguments(boolean... arguments)
OpContext
setBArguments
in interface OpContext
public List<Boolean> getBArguments()
getBArguments
in interface OpContext
public void setInputArray(int index, @NonNull INDArray array)
OpContext
setInputArray
in interface OpContext
public List<INDArray> getInputArrays()
OpContext
getInputArrays
in interface OpContext
public List<INDArray> getOutputArrays()
OpContext
getOutputArrays
in interface OpContext
public void setOutputArray(int index, @NonNull INDArray array)
OpContext
setOutputArray
in interface OpContext
public void setInputArrays(@NonNull List<INDArray> arrays)
OpContext
setInputArrays
in interface OpContext
public void setOutputArrays(@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 © 2019. All rights reserved.