| Package | Description |
|---|---|
| org.nd4j.autodiff.samediff.internal |
| Modifier and Type | Field and Description |
|---|---|
protected FrameIter |
AbstractSession.ExecStepPredicate.currParentFrame |
protected FrameIter |
AbstractSession.ExecStep.frameIter |
protected FrameIter |
InferenceSession.Dep.parentFrame |
| Modifier and Type | Method and Description |
|---|---|
FrameIter |
FrameIter.clone() |
FrameIter |
AbstractSession.VarId.toFrameIter() |
| Modifier and Type | Method and Description |
|---|---|
protected void |
AbstractSession.addDependenciesForOp(String opName,
FrameIter depFrameIter)
Suppose operation X has just been executed.
|
boolean |
AbstractSession.contains(String variable,
String frame,
int iteration,
FrameIter parentFrameIter) |
INDArray[] |
InferenceSession.doExec(DifferentialFunction op,
OpContext opContext,
FrameIter outputFrameIter,
Set<AbstractSession.VarId> opInputs,
Set<AbstractSession.VarId> allIterInputs,
Set<String> constAndPhInputs) |
T |
AbstractSession.get(String variable,
String frame,
int iteration,
FrameIter parentFrameIter)
Get a previously calculated output; throws an exception if the output does not exist
|
T |
AbstractSession.get(String variable,
String frame,
int iteration,
FrameIter parentFrameIter,
boolean enforceExistence)
Get a previously calculated output
|
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
|
protected AbstractSession.ExecStep |
AbstractSession.getExecStepForVar(String varName,
FrameIter frameIter)
Get the ExecStep for the given variable, given execution is happening at the specified frame/iteration
|
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[] |
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
|
protected void |
AbstractSession.updateDescendantDeps(AbstractSession.ExecStep justExecuted,
FrameIter outFrameIter)
Update the descendant dependencies
So if the graph structure is X -> A, then add all (X,Y,Z,...) -> A to the dependency tracker
This is for a specific frame and iteration, for both sides of the dependency (in and out)
|
| Constructor and Description |
|---|
ExecStep(@NonNull AbstractSession.ExecType execType,
@NonNull String name,
FrameIter frameIter) |
OpDep(@NonNull String opName,
@NonNull String frame,
int iter,
FrameIter parentFrame) |
Copyright © 2021. All rights reserved.