public class DataTypesSession extends AbstractSession<DataType,DataTypesSession.DataTypeCalc>
Modifier and Type | Class and Description |
---|---|
protected static class |
DataTypesSession.DataTypeCalc |
AbstractSession.FrameIter, AbstractSession.VarId
availableForExec, availableForExecSet, execConstInputs, execInputs, execInputsAllIter, frameParents, nodeOutputs, OUTER_FRAME, sameDiff, subgraph, tensorArrays
Constructor and Description |
---|
DataTypesSession(SameDiff sameDiff) |
Modifier and Type | Method and Description |
---|---|
DataTypesSession.DataTypeCalc |
getAndParameterizeOp(String opName,
AbstractSession.FrameIter frameIter,
Set<AbstractSession.VarId> inputs,
Set<AbstractSession.VarId> allIterInputs,
Set<String> constAndPhInputs,
Map<String,DataType> placeholderValues)
Get the parameterized op to execute - for example, the op/DifferentialFunction with all inputs set
|
DataType |
getConstantOrVariable(String variableName)
Get the constant or variable output - for example, constant array or constant shape.
|
DataType[] |
getOutputs(DataTypesSession.DataTypeCalc op,
AbstractSession.FrameIter outputFrameIter,
Set<AbstractSession.VarId> inputs,
Set<AbstractSession.VarId> allIterInputs,
Set<String> constAndPhInputs)
Execute the op - calculate INDArrays, or shape info, etc
|
addToExecInputs, allInputsAvailable, contains, get, get, initSubgraph, lookup, newVarId, newVarId, output, preprocessPlaceholders, updateDescendentsForExec
public DataTypesSession(SameDiff sameDiff)
public DataType getConstantOrVariable(String variableName)
AbstractSession
getConstantOrVariable
in class AbstractSession<DataType,DataTypesSession.DataTypeCalc>
variableName
- The name of the variable to get the constant forpublic DataTypesSession.DataTypeCalc getAndParameterizeOp(String opName, AbstractSession.FrameIter frameIter, Set<AbstractSession.VarId> inputs, Set<AbstractSession.VarId> allIterInputs, Set<String> constAndPhInputs, Map<String,DataType> placeholderValues)
AbstractSession
getAndParameterizeOp
in class AbstractSession<DataType,DataTypesSession.DataTypeCalc>
opName
- Name of the opframeIter
- The frame and iteration of the op outputsinputs
- The inputs to the op (excluding constants/placeholders) - for the specific frame + iterationallIterInputs
- The inputs - those that are not iteration-specific (mainly Enter op vars, which might be used in all iterations but are only executed once on iter 0)constAndPhInputs
- The constant and placeholder inputs - used for all frames/iterationspublic DataType[] getOutputs(DataTypesSession.DataTypeCalc op, AbstractSession.FrameIter outputFrameIter, Set<AbstractSession.VarId> inputs, Set<AbstractSession.VarId> allIterInputs, Set<String> constAndPhInputs)
AbstractSession
getOutputs
in class AbstractSession<DataType,DataTypesSession.DataTypeCalc>
op
- Operation to exit. This should be parameterized (i.e., all inputs set)outputFrameIter
- The frame and iteration of the outputsinputs
- The specific input arrays for the opCopyright © 2019. All rights reserved.