| Package | Description |
|---|---|
| org.nd4j.autodiff.samediff.internal |
| Modifier and Type | Field and Description |
|---|---|
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)
Get the VarId from the specified name.
|
protected static AbstractSession.VarId |
AbstractSession.lookup(String name,
Collection<AbstractSession.VarId> varIds,
Collection<AbstractSession.VarId> varIds2,
boolean exceptionOnNotFound)
Get the VarId from the specified name.
|
protected AbstractSession.VarId |
AbstractSession.ExecStep.toVarId() |
AbstractSession.VarId |
FrameIter.toVarId(String name) |
| Modifier and Type | Method and Description |
|---|---|
INDArray[] |
InferenceSession.doExec(DifferentialFunction op,
OpContext opContext,
FrameIter outputFrameIter,
Set<AbstractSession.VarId> opInputs,
Set<AbstractSession.VarId> allIterInputs,
Set<String> constAndPhInputs) |
INDArray[] |
InferenceSession.doExec(DifferentialFunction op,
OpContext opContext,
FrameIter outputFrameIter,
Set<AbstractSession.VarId> opInputs,
Set<AbstractSession.VarId> allIterInputs,
Set<String> constAndPhInputs) |
Pair<SameDiffOp,OpContext> |
InferenceSession.getAndParameterizeOp(String opName,
FrameIter frameIter,
Set<AbstractSession.VarId> opInputs,
Set<AbstractSession.VarId> allIterInputs,
Set<String> constAndPhInputs,
Map<String,INDArray> placeholderValues,
Set<String> allReqVariables) |
Pair<SameDiffOp,OpContext> |
InferenceSession.getAndParameterizeOp(String opName,
FrameIter frameIter,
Set<AbstractSession.VarId> opInputs,
Set<AbstractSession.VarId> allIterInputs,
Set<String> constAndPhInputs,
Map<String,INDArray> placeholderValues,
Set<String> allReqVariables) |
abstract O |
AbstractSession.getAndParameterizeOp(String opName,
FrameIter frameIter,
Set<AbstractSession.VarId> inputs,
Set<AbstractSession.VarId> allIterInputs,
Set<String> constAndPhInputs,
Map<String,T> placeholderValues,
Set<String> allReqVariables)
Get the parameterized op to execute - for example, the op/DifferentialFunction with all inputs set
|
abstract O |
AbstractSession.getAndParameterizeOp(String opName,
FrameIter frameIter,
Set<AbstractSession.VarId> inputs,
Set<AbstractSession.VarId> allIterInputs,
Set<String> constAndPhInputs,
Map<String,T> placeholderValues,
Set<String> allReqVariables)
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) |
abstract T[] |
AbstractSession.getOutputs(O op,
FrameIter outputFrameIter,
Set<AbstractSession.VarId> inputs,
Set<AbstractSession.VarId> allIterInputs,
Set<String> constAndPhInputs,
List<Listener> listeners,
At at,
MultiDataSet batch,
Set<String> allReqVariables)
Execute the op - calculate INDArrays, or shape info, etc
|
abstract T[] |
AbstractSession.getOutputs(O op,
FrameIter outputFrameIter,
Set<AbstractSession.VarId> inputs,
Set<AbstractSession.VarId> allIterInputs,
Set<String> constAndPhInputs,
List<Listener> listeners,
At at,
MultiDataSet batch,
Set<String> allReqVariables)
Execute the op - calculate INDArrays, or shape info, etc
|
INDArray[] |
InferenceSession.getOutputs(Pair<SameDiffOp,OpContext> opPair,
FrameIter outputFrameIter,
Set<AbstractSession.VarId> opInputs,
Set<AbstractSession.VarId> allIterInputs,
Set<String> constAndPhInputs,
List<Listener> listeners,
At at,
MultiDataSet batch,
Set<String> allReqVariables) |
INDArray[] |
InferenceSession.getOutputs(Pair<SameDiffOp,OpContext> opPair,
FrameIter outputFrameIter,
Set<AbstractSession.VarId> opInputs,
Set<AbstractSession.VarId> allIterInputs,
Set<String> constAndPhInputs,
List<Listener> listeners,
At at,
MultiDataSet batch,
Set<String> allReqVariables) |
INDArray[] |
TrainingSession.getOutputs(Pair<SameDiffOp,OpContext> opPair,
FrameIter outputFrameIter,
Set<AbstractSession.VarId> opInputs,
Set<AbstractSession.VarId> allIterInputs,
Set<String> constAndPhInputs,
List<Listener> listeners,
At at,
MultiDataSet batch,
Set<String> allReqVariables) |
INDArray[] |
TrainingSession.getOutputs(Pair<SameDiffOp,OpContext> opPair,
FrameIter outputFrameIter,
Set<AbstractSession.VarId> opInputs,
Set<AbstractSession.VarId> allIterInputs,
Set<String> constAndPhInputs,
List<Listener> listeners,
At at,
MultiDataSet batch,
Set<String> allReqVariables) |
INDArray[] |
InferenceSession.getOutputsHelperTensorArrayOps(DifferentialFunction op,
FrameIter outputFrameIter,
Set<AbstractSession.VarId> opInputs,
Set<AbstractSession.VarId> allIterInputs)
Forward pass for TensorArray ops
|
INDArray[] |
InferenceSession.getOutputsHelperTensorArrayOps(DifferentialFunction op,
FrameIter outputFrameIter,
Set<AbstractSession.VarId> opInputs,
Set<AbstractSession.VarId> allIterInputs)
Forward pass for TensorArray ops
|
protected static AbstractSession.VarId |
AbstractSession.lookup(String name,
Collection<AbstractSession.VarId> varIds,
boolean exceptionOnNotFound)
Get the VarId from the specified name.
|
protected static AbstractSession.VarId |
AbstractSession.lookup(String name,
Collection<AbstractSession.VarId> varIds,
Collection<AbstractSession.VarId> varIds2,
boolean exceptionOnNotFound)
Get the VarId from the specified name.
|
protected static AbstractSession.VarId |
AbstractSession.lookup(String name,
Collection<AbstractSession.VarId> varIds,
Collection<AbstractSession.VarId> varIds2,
boolean exceptionOnNotFound)
Get the VarId from the specified name.
|
Copyright © 2021. All rights reserved.