| Package | Description |
|---|---|
| org.nd4j.autodiff.samediff.internal |
| Modifier and Type | Field and Description |
|---|---|
protected Queue<AbstractSession.VarId> |
AbstractSession.availableForExec |
protected Set<AbstractSession.VarId> |
AbstractSession.availableForExecSet |
protected Map<AbstractSession.VarId,Set<AbstractSession.VarId>> |
AbstractSession.execInputs
Stores what variables are required to calculate the specific variable.
|
protected Map<AbstractSession.VarId,Set<AbstractSession.VarId>> |
AbstractSession.execInputs
Stores what variables are required to calculate the specific variable.
|
protected Map<AbstractSession.VarId,Set<AbstractSession.VarId>> |
AbstractSession.execInputsAllIter
As per execInputs map - with the different that the iteration number should be ignored (i.e., always 0)
Reason: Enter nodes - these are executed once
Example: EnterOp(x) -> LoopCondition(less(x,y)): less op requires "X" on all iterations which is the output of the
enter op, which is only executed for iteration 0 in a frame.
|
protected Map<AbstractSession.VarId,Set<AbstractSession.VarId>> |
AbstractSession.execInputsAllIter
As per execInputs map - with the different that the iteration number should be ignored (i.e., always 0)
Reason: Enter nodes - these are executed once
Example: EnterOp(x) -> LoopCondition(less(x,y)): less op requires "X" on all iterations which is the output of the
enter op, which is only executed for iteration 0 in a frame.
|
protected Map<AbstractSession.VarId,T> |
AbstractSession.nodeOutputs |
protected Map<AbstractSession.VarId,List<T>> |
AbstractSession.tensorArrays |
| Modifier and Type | Method and Description |
|---|---|
protected static AbstractSession.VarId |
AbstractSession.lookup(String name,
Collection<AbstractSession.VarId> varIds,
boolean exceptionOnNotFound) |
AbstractSession.VarId |
AbstractSession.newVarId(String variable,
AbstractSession.FrameIter frameIter) |
AbstractSession.VarId |
AbstractSession.newVarId(String variable,
String frame,
int iteration,
AbstractSession.FrameIter parentFrameIter) |
| Modifier and Type | Method and Description |
|---|---|
protected void |
AbstractSession.addToExecInputs(boolean isConstOrPh,
AbstractSession.VarId inputVar,
AbstractSession.VarId forVariable)
This method is used to record that the specified input is required for calculating the specified output.
|
protected boolean |
AbstractSession.allInputsAvailable(int execStep,
String[] inputsThisOp,
AbstractSession.VarId executedVar) |
protected void |
AbstractSession.updateDescendentsForExec(int execStep,
AbstractSession.VarId executedVar)
This method should be called for a variable once it's array is ready for use.
|
| Modifier and Type | Method and Description |
|---|---|
DataTypesSession.DataTypeCalc |
DataTypesSession.getAndParameterizeOp(String opName,
AbstractSession.FrameIter frameIter,
Set<AbstractSession.VarId> inputs,
Set<AbstractSession.VarId> allIterInputs,
Set<String> constAndPhInputs,
Map<String,DataType> placeholderValues) |
DataTypesSession.DataTypeCalc |
DataTypesSession.getAndParameterizeOp(String opName,
AbstractSession.FrameIter frameIter,
Set<AbstractSession.VarId> inputs,
Set<AbstractSession.VarId> allIterInputs,
Set<String> constAndPhInputs,
Map<String,DataType> placeholderValues) |
DifferentialFunction |
InferenceSession.getAndParameterizeOp(String opName,
AbstractSession.FrameIter frameIter,
Set<AbstractSession.VarId> opInputs,
Set<AbstractSession.VarId> allIterInputs,
Set<String> constAndPhInputs,
Map<String,INDArray> placeholderValues) |
DifferentialFunction |
InferenceSession.getAndParameterizeOp(String opName,
AbstractSession.FrameIter frameIter,
Set<AbstractSession.VarId> opInputs,
Set<AbstractSession.VarId> allIterInputs,
Set<String> constAndPhInputs,
Map<String,INDArray> placeholderValues) |
abstract O |
AbstractSession.getAndParameterizeOp(String opName,
AbstractSession.FrameIter frameIter,
Set<AbstractSession.VarId> inputs,
Set<AbstractSession.VarId> allIterInputs,
Set<String> constAndPhInputs,
Map<String,T> placeholderValues)
Get the parameterized op to execute - for example, the op/DifferentialFunction with all inputs set
|
abstract O |
AbstractSession.getAndParameterizeOp(String opName,
AbstractSession.FrameIter frameIter,
Set<AbstractSession.VarId> inputs,
Set<AbstractSession.VarId> allIterInputs,
Set<String> constAndPhInputs,
Map<String,T> placeholderValues)
Get the parameterized op to execute - for example, the op/DifferentialFunction with all inputs set
|
protected INDArray |
InferenceSession.getArray(SDVariable sdv,
Collection<AbstractSession.VarId> opInputs,
Collection<AbstractSession.VarId> allIterInputs) |
protected INDArray |
InferenceSession.getArray(SDVariable sdv,
Collection<AbstractSession.VarId> opInputs,
Collection<AbstractSession.VarId> allIterInputs) |
DataType[] |
DataTypesSession.getOutputs(DataTypesSession.DataTypeCalc op,
AbstractSession.FrameIter outputFrameIter,
Set<AbstractSession.VarId> inputs,
Set<AbstractSession.VarId> allIterInputs,
Set<String> constAndPhInputs,
List<Listener> listeners,
At at,
MultiDataSet batch) |
DataType[] |
DataTypesSession.getOutputs(DataTypesSession.DataTypeCalc op,
AbstractSession.FrameIter outputFrameIter,
Set<AbstractSession.VarId> inputs,
Set<AbstractSession.VarId> allIterInputs,
Set<String> constAndPhInputs,
List<Listener> listeners,
At at,
MultiDataSet batch) |
INDArray[] |
InferenceSession.getOutputs(DifferentialFunction op,
AbstractSession.FrameIter outputFrameIter,
Set<AbstractSession.VarId> opInputs,
Set<AbstractSession.VarId> allIterInputs,
Set<String> constAndPhInputs,
List<Listener> listeners,
At at,
MultiDataSet batch) |
INDArray[] |
InferenceSession.getOutputs(DifferentialFunction op,
AbstractSession.FrameIter outputFrameIter,
Set<AbstractSession.VarId> opInputs,
Set<AbstractSession.VarId> allIterInputs,
Set<String> constAndPhInputs,
List<Listener> listeners,
At at,
MultiDataSet batch) |
abstract T[] |
AbstractSession.getOutputs(O op,
AbstractSession.FrameIter outputFrameIter,
Set<AbstractSession.VarId> inputs,
Set<AbstractSession.VarId> allIterInputs,
Set<String> constAndPhInputs,
List<Listener> listeners,
At at,
MultiDataSet batch)
Execute the op - calculate INDArrays, or shape info, etc
|
abstract T[] |
AbstractSession.getOutputs(O op,
AbstractSession.FrameIter outputFrameIter,
Set<AbstractSession.VarId> inputs,
Set<AbstractSession.VarId> allIterInputs,
Set<String> constAndPhInputs,
List<Listener> listeners,
At at,
MultiDataSet batch)
Execute the op - calculate INDArrays, or shape info, etc
|
INDArray[] |
InferenceSession.getOutputsHelper(DifferentialFunction op,
AbstractSession.FrameIter outputFrameIter,
Set<AbstractSession.VarId> opInputs,
Set<AbstractSession.VarId> allIterInputs,
Set<String> constAndPhInputs) |
INDArray[] |
InferenceSession.getOutputsHelper(DifferentialFunction op,
AbstractSession.FrameIter outputFrameIter,
Set<AbstractSession.VarId> opInputs,
Set<AbstractSession.VarId> allIterInputs,
Set<String> constAndPhInputs) |
protected static AbstractSession.VarId |
AbstractSession.lookup(String name,
Collection<AbstractSession.VarId> varIds,
boolean exceptionOnNotFound) |
Copyright © 2019. All rights reserved.