public interface OpContext extends AutoCloseable
Modifier and Type | Method and Description |
---|---|
void |
allowHelpers(boolean reallyAllow)
This method allows to enable/disable use of platform helpers within ops.
|
org.bytedeco.javacpp.Pointer |
contextPointer()
This method returns pointer to context, to be used during native op execution
|
List<Boolean> |
getBArguments() |
List<DataType> |
getDArguments() |
ExecutionMode |
getExecutionMode()
This method returns current execution mode for Context
|
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
|
Pair<Long,Long> |
getRngStates()
This method returns RNG states, root first node second
|
List<Double> |
getTArguments() |
void |
markInplace(boolean reallyInplace)
This method allows to set op as inplace
|
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 |
setExecutionMode(ExecutionMode mode)
This method allows to set certain execution mode
|
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 |
setRngStates(long rootState,
long nodeState)
This method sets root-level seed for rng
|
void |
setTArguments(double... arguments)
This method sets floating point arguments required for operation
|
void |
shapeFunctionOverride(boolean reallyOverride)
This methos allows to disape outputs validation via shape function
|
close
void setIArguments(long... arguments)
arguments
- int numIArguments()
void setTArguments(double... arguments)
arguments
- int numTArguments()
void setDArguments(DataType... arguments)
arguments
- int numDArguments()
void setBArguments(boolean... arguments)
arguments
- int numBArguments()
void setRngStates(long rootState, long nodeState)
seed
- Pair<Long,Long> getRngStates()
void setInputArray(int index, INDArray array)
index
- array
- void setInputArrays(List<INDArray> arrays)
arrays
- void setInputArrays(INDArray... arrays)
arrays
- List<INDArray> getInputArrays()
int numInputArguments()
INDArray getInputArray(int idx)
void setOutputArray(int index, INDArray array)
index
- array
- void setOutputArrays(List<INDArray> arrays)
arrays
- void setOutputArrays(INDArray... arrays)
arrays
- List<INDArray> getOutputArrays()
INDArray getOutputArray(int i)
int numOutputArguments()
org.bytedeco.javacpp.Pointer contextPointer()
void markInplace(boolean reallyInplace)
reallyInplace
- void allowHelpers(boolean reallyAllow)
reallyAllow
- void shapeFunctionOverride(boolean reallyOverride)
reallyOverride
- ExecutionMode getExecutionMode()
void setExecutionMode(ExecutionMode mode)
mode
- void purge()
Copyright © 2020. All rights reserved.