Modifier and Type | Method and Description |
---|---|
List<LongShapeDescriptor> |
DifferentialFunction.calculateOutputShape(OpContext oc) |
Modifier and Type | Method and Description |
---|---|
void |
BaseListener.opExecution(SameDiff sd,
At at,
MultiDataSet batch,
SameDiffOp op,
OpContext opContext,
INDArray[] outputs) |
void |
Listener.opExecution(SameDiff sd,
At at,
MultiDataSet batch,
SameDiffOp op,
OpContext opContext,
INDArray[] outputs)
Called at the end of each operation execution
|
void |
BaseListener.preOpExecution(SameDiff sd,
At at,
SameDiffOp op,
OpContext opContext) |
void |
Listener.preOpExecution(SameDiff sd,
At at,
SameDiffOp op,
OpContext opContext)
Called just before each operation is executed (native code called, etc) - after all inputs etc have been set
|
Modifier and Type | Method and Description |
---|---|
void |
OpBenchmarkListener.opExecution(SameDiff sd,
At at,
MultiDataSet batch,
SameDiffOp op,
OpContext opContext,
INDArray[] outputs) |
void |
ArraySavingListener.opExecution(SameDiff sd,
At at,
MultiDataSet batch,
SameDiffOp op,
OpContext opContext,
INDArray[] outputs) |
void |
OpBenchmarkListener.preOpExecution(SameDiff sd,
At at,
SameDiffOp op,
OpContext opContext) |
void |
ExecDebuggingListener.preOpExecution(SameDiff sd,
At at,
SameDiffOp op,
OpContext opContext) |
Modifier and Type | Method and Description |
---|---|
void |
UIListener.opExecution(SameDiff sd,
At at,
MultiDataSet batch,
SameDiffOp op,
OpContext opContext,
INDArray[] outputs) |
Modifier and Type | Method and Description |
---|---|
void |
ProfilingListener.opExecution(SameDiff sd,
At at,
MultiDataSet batch,
SameDiffOp op,
OpContext opContext,
INDArray[] outputs) |
void |
ProfilingListener.preOpExecution(SameDiff sd,
At at,
SameDiffOp op,
OpContext opContext) |
Modifier and Type | Method and Description |
---|---|
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) |
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) |
Modifier and Type | Method and Description |
---|---|
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.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) |
Modifier and Type | Method and Description |
---|---|
void |
ActivationGradientCheckListener.opExecution(SameDiff sd,
At at,
MultiDataSet batch,
SameDiffOp op,
OpContext opContext,
INDArray[] outputs) |
Modifier and Type | Method and Description |
---|---|
void |
NonInplaceValidationListener.opExecution(SameDiff sd,
At at,
MultiDataSet batch,
SameDiffOp op,
OpContext opContext,
INDArray[] outputs) |
void |
NonInplaceValidationListener.preOpExecution(SameDiff sd,
At at,
SameDiffOp op,
OpContext oc) |
Modifier and Type | Class and Description |
---|---|
class |
BaseOpContext
Implementation of common methods for OpContext
|
Modifier and Type | Method and Description |
---|---|
List<LongShapeDescriptor> |
ScatterUpdate.calculateOutputShape(OpContext opContext) |
Modifier and Type | Method and Description |
---|---|
OpContext |
DefaultOpExecutioner.buildContext() |
OpContext |
OpExecutioner.buildContext()
This method returns OpContext which can be used (and reused) to execute custom ops
|
Modifier and Type | Method and Description |
---|---|
List<LongShapeDescriptor> |
DefaultOpExecutioner.calculateOutputShape(CustomOp op,
OpContext opContext) |
List<LongShapeDescriptor> |
OpExecutioner.calculateOutputShape(CustomOp op,
OpContext opContext) |
static void |
OpExecutionerUtil.checkForInf(CustomOp op,
OpContext oc) |
static void |
OpExecutionerUtil.checkForInf(Op op,
OpContext oc) |
static void |
OpExecutionerUtil.checkForNaN(CustomOp op,
OpContext oc) |
static void |
OpExecutionerUtil.checkForNaN(Op op,
OpContext oc) |
protected void |
DefaultOpExecutioner.checkForWorkspaces(CustomOp op,
OpContext oc) |
protected void |
DefaultOpExecutioner.checkForWorkspaces(Op op,
OpContext oc) |
INDArray[] |
DefaultOpExecutioner.exec(CustomOp op,
OpContext context) |
INDArray[] |
OpExecutioner.exec(CustomOp op,
OpContext context)
This method executes op with given context
|
abstract INDArray |
DefaultOpExecutioner.exec(Op op,
OpContext opContext) |
INDArray |
OpExecutioner.exec(Op op,
OpContext opContext)
Execute the operation
|
INDArray |
DefaultOpExecutioner.getX(Op op,
OpContext oc) |
INDArray |
DefaultOpExecutioner.getY(Op op,
OpContext oc) |
INDArray |
DefaultOpExecutioner.getZ(Op op,
OpContext oc) |
long |
DefaultOpExecutioner.profilingConfigurableHookIn(CustomOp op,
OpContext oc) |
void |
DefaultOpExecutioner.profilingConfigurableHookOut(CustomOp op,
OpContext oc,
long timeStart) |
void |
DefaultOpExecutioner.profilingConfigurableHookOut(Op op,
OpContext oc,
long timeStart) |
long |
DefaultOpExecutioner.profilingHookIn(CustomOp op,
OpContext oc)
Deprecated.
|
void |
DefaultOpExecutioner.profilingHookOut(CustomOp op,
OpContext oc,
long timeStart)
Deprecated.
|
void |
DefaultOpExecutioner.profilingHookOut(Op op,
OpContext oc,
long timeStart)
Deprecated.
|
void |
DefaultOpExecutioner.setX(INDArray x,
Op op,
OpContext oc) |
void |
DefaultOpExecutioner.setY(INDArray y,
Op op,
OpContext oc) |
void |
DefaultOpExecutioner.setZ(INDArray z,
Op op,
OpContext oc) |
Modifier and Type | Method and Description |
---|---|
List<LongShapeDescriptor> |
ExternalErrorsFunction.calculateOutputShape(OpContext oc) |
Modifier and Type | Method and Description |
---|---|
List<LongShapeDescriptor> |
Variance.calculateOutputShape(OpContext oc) |
DataType |
Variance.resultType(OpContext oc) |
boolean |
Variance.validateDataTypes(OpContext oc) |
Modifier and Type | Method and Description |
---|---|
DataType |
MaxOut.resultType(OpContext oc) |
boolean |
MaxOut.validateDataTypes(OpContext oc,
boolean experimentalMode) |
Modifier and Type | Method and Description |
---|---|
List<LongShapeDescriptor> |
BaseRandomOp.calculateOutputShape(OpContext opContext) |
Modifier and Type | Method and Description |
---|---|
static INDArray[] |
Nd4j.exec(CustomOp op,
OpContext context)
Execute the operation and return the result
|
static INDArray |
Nd4j.exec(Op op,
OpContext context) |
Copyright © 2020. All rights reserved.