Package | Description |
---|---|
org.nd4j.autodiff.samediff.internal |
Modifier and Type | Field and Description |
---|---|
protected DependencyTracker<AbstractSession.ExecStep,AbstractSession.ExecStep> |
AbstractSession.dt |
protected DependencyTracker<AbstractSession.ExecStep,AbstractSession.ExecStep> |
AbstractSession.dt |
Modifier and Type | Method and Description |
---|---|
protected AbstractSession.ExecStep |
AbstractSession.getExecStepForVar(String varName,
AbstractSession.FrameIter frameIter)
Get the ExecStep for the given variable, given execution is happening at the specified frame/iteration
|
Modifier and Type | Method and Description |
---|---|
protected void |
AbstractSession.addVarControlDeps(AbstractSession.ExecStep es,
Variable v)
Add the control dependency from Op -> variable
|
boolean |
AbstractSession.ExecStepPredicate.test(AbstractSession.ExecStep execStep) |
protected void |
AbstractSession.updateDescendantDeps(AbstractSession.ExecStep justExecuted,
AbstractSession.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)
|
Copyright © 2019. All rights reserved.