Uses of Interface
org.nd4j.linalg.api.ops.OpContext
-
-
Uses of OpContext in org.nd4j.autodiff.functions
Methods in org.nd4j.autodiff.functions with parameters of type OpContext Modifier and Type Method Description List<LongShapeDescriptor>DifferentialFunction. calculateOutputShape(OpContext oc) -
Uses of OpContext in org.nd4j.autodiff.listeners
Methods in org.nd4j.autodiff.listeners with parameters of type OpContext Modifier and Type Method Description voidBaseListener. opExecution(SameDiff sd, At at, MultiDataSet batch, SameDiffOp op, OpContext opContext, INDArray[] outputs)voidListener. opExecution(SameDiff sd, At at, MultiDataSet batch, SameDiffOp op, OpContext opContext, INDArray[] outputs)Called at the end of each operation executionvoidBaseListener. preOpExecution(SameDiff sd, At at, SameDiffOp op, OpContext opContext)voidListener. 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 -
Uses of OpContext in org.nd4j.autodiff.listeners.debugging
Methods in org.nd4j.autodiff.listeners.debugging with parameters of type OpContext Modifier and Type Method Description voidArraySavingListener. opExecution(SameDiff sd, At at, MultiDataSet batch, SameDiffOp op, OpContext opContext, INDArray[] outputs)voidControlflowListener. opExecution(SameDiff sd, At at, MultiDataSet batch, SameDiffOp op, OpContext opContext, INDArray[] outputs)voidOpBenchmarkListener. opExecution(SameDiff sd, At at, MultiDataSet batch, SameDiffOp op, OpContext opContext, INDArray[] outputs)voidControlflowListener. preOpExecution(SameDiff sd, At at, SameDiffOp op, OpContext opContext)voidExecDebuggingListener. preOpExecution(SameDiff sd, At at, SameDiffOp op, OpContext opContext)voidOpBenchmarkListener. preOpExecution(SameDiff sd, At at, SameDiffOp op, OpContext opContext) -
Uses of OpContext in org.nd4j.autodiff.listeners.impl
Methods in org.nd4j.autodiff.listeners.impl with parameters of type OpContext Modifier and Type Method Description voidUIListener. opExecution(SameDiff sd, At at, MultiDataSet batch, SameDiffOp op, OpContext opContext, INDArray[] outputs) -
Uses of OpContext in org.nd4j.autodiff.listeners.profiler
Methods in org.nd4j.autodiff.listeners.profiler with parameters of type OpContext Modifier and Type Method Description voidProfilingListener. opExecution(SameDiff sd, At at, MultiDataSet batch, SameDiffOp op, OpContext opContext, INDArray[] outputs)voidProfilingListener. preOpExecution(SameDiff sd, At at, SameDiffOp op, OpContext opContext) -
Uses of OpContext in org.nd4j.autodiff.samediff.config
Methods in org.nd4j.autodiff.samediff.config with parameters of type OpContext Modifier and Type Method Description static ExecutionResultExecutionResult. createFrom(DifferentialFunction func, OpContext opContext) -
Uses of OpContext in org.nd4j.autodiff.samediff.internal
Methods in org.nd4j.autodiff.samediff.internal that return types with arguments of type OpContext Modifier and Type Method 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, Map<String,SDValue> otherPlaceholders)Methods in org.nd4j.autodiff.samediff.internal with parameters of type OpContext Modifier and Type Method Description ExecutionResultInferenceSession. doExec(DifferentialFunction op, OpContext opContext, FrameIter outputFrameIter, Set<AbstractSession.VarId> opInputs, Set<AbstractSession.VarId> allIterInputs, Set<String> constAndPhInputs, Map<String,SDValue> otherPlaceHolders)Method parameters in org.nd4j.autodiff.samediff.internal with type arguments of type OpContext Modifier and Type Method Description ExecutionResultInferenceSession. 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, Map<String,SDValue> otherPlaceHolders)ExecutionResultTrainingSession. 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, Map<String,SDValue> otherPlaceHolders) -
Uses of OpContext in org.nd4j.autodiff.validation
Methods in org.nd4j.autodiff.validation with parameters of type OpContext Modifier and Type Method Description voidActivationGradientCheckListener. opExecution(SameDiff sd, At at, MultiDataSet batch, SameDiffOp op, OpContext opContext, INDArray[] outputs) -
Uses of OpContext in org.nd4j.linalg.api.ops
Classes in org.nd4j.linalg.api.ops that implement OpContext Modifier and Type Class Description classBaseOpContext -
Uses of OpContext in org.nd4j.linalg.api.ops.compat
Methods in org.nd4j.linalg.api.ops.compat with parameters of type OpContext Modifier and Type Method Description List<LongShapeDescriptor>CompatSparseToDense. calculateOutputShape(OpContext oc) -
Uses of OpContext in org.nd4j.linalg.api.ops.custom
Methods in org.nd4j.linalg.api.ops.custom with parameters of type OpContext Modifier and Type Method Description List<LongShapeDescriptor>Invoke. calculateOutputShape(OpContext oc) -
Uses of OpContext in org.nd4j.linalg.api.ops.executioner
Methods in org.nd4j.linalg.api.ops.executioner that return OpContext Modifier and Type Method Description OpContextDefaultOpExecutioner. buildContext()OpContextOpExecutioner. buildContext()This method returns OpContext which can be used (and reused) to execute custom opsMethods in org.nd4j.linalg.api.ops.executioner with parameters of type OpContext Modifier and Type Method Description List<LongShapeDescriptor>DefaultOpExecutioner. calculateOutputShape(CustomOp op, OpContext opContext)List<LongShapeDescriptor>OpExecutioner. calculateOutputShape(CustomOp op, OpContext opContext)static voidOpExecutionerUtil. checkForInf(CustomOp op, OpContext oc)static voidOpExecutionerUtil. checkForInf(Op op, OpContext oc)static voidOpExecutionerUtil. checkForNaN(CustomOp op, OpContext oc)static voidOpExecutionerUtil. checkForNaN(Op op, OpContext oc)protected voidDefaultOpExecutioner. checkForWorkspaces(CustomOp op, OpContext oc)protected voidDefaultOpExecutioner. checkForWorkspaces(Op op, OpContext oc)INDArray[]DefaultOpExecutioner. exec(CustomOp op, OpContext context)abstract INDArrayDefaultOpExecutioner. exec(Op op, OpContext opContext)INDArray[]OpExecutioner. exec(CustomOp op, OpContext context)This method executes op with given contextINDArrayOpExecutioner. exec(Op op, OpContext opContext)Execute the operationINDArrayDefaultOpExecutioner. getX(Op op, OpContext oc)INDArrayDefaultOpExecutioner. getY(Op op, OpContext oc)INDArrayDefaultOpExecutioner. getZ(Op op, OpContext oc)longDefaultOpExecutioner. profilingConfigurableHookIn(CustomOp op, OpContext oc)voidDefaultOpExecutioner. profilingConfigurableHookOut(CustomOp op, OpContext oc, long timeStart)voidDefaultOpExecutioner. profilingConfigurableHookOut(Op op, OpContext oc, long timeStart)longDefaultOpExecutioner. profilingHookIn(CustomOp op, OpContext oc)Deprecated.voidDefaultOpExecutioner. profilingHookOut(CustomOp op, OpContext oc, long timeStart)Deprecated.voidDefaultOpExecutioner. profilingHookOut(Op op, OpContext oc, long timeStart)Deprecated.voidDefaultOpExecutioner. setX(INDArray x, Op op, OpContext oc)voidDefaultOpExecutioner. setY(INDArray y, Op op, OpContext oc)voidDefaultOpExecutioner. setZ(INDArray z, Op op, OpContext oc) -
Uses of OpContext in org.nd4j.linalg.api.ops.impl.layers
Methods in org.nd4j.linalg.api.ops.impl.layers with parameters of type OpContext Modifier and Type Method Description List<LongShapeDescriptor>ExternalErrorsFunction. calculateOutputShape(OpContext oc) -
Uses of OpContext in org.nd4j.linalg.api.ops.impl.summarystats
Methods in org.nd4j.linalg.api.ops.impl.summarystats with parameters of type OpContext Modifier and Type Method Description List<LongShapeDescriptor>Variance. calculateOutputShape(OpContext oc)DataTypeVariance. resultType(OpContext oc)booleanVariance. validateDataTypes(OpContext oc) -
Uses of OpContext in org.nd4j.linalg.api.ops.impl.transforms
Methods in org.nd4j.linalg.api.ops.impl.transforms with parameters of type OpContext Modifier and Type Method Description DataTypeMaxOut. resultType(OpContext oc)booleanMaxOut. validateDataTypes(OpContext oc, boolean experimentalMode) -
Uses of OpContext in org.nd4j.linalg.api.ops.random.impl
-
Uses of OpContext in org.nd4j.linalg.factory
Methods in org.nd4j.linalg.factory with parameters of type OpContext Modifier and Type Method Description static INDArray[]Nd4j. exec(CustomOp op, OpContext context)Execute the operation and return the resultstatic INDArrayNd4j. exec(Op op, OpContext context)
-