public abstract class BaseArithmeticBackpropOp extends BaseDynamicTransformOp
DynamicCustomOp.DynamicCustomOpsBuilder, DynamicCustomOp.SameDiffBuilderiArguments, inplaceCall, inputArguments, outputArguments, outputVariables, tArgumentsdimensions, extraArgs, inPlace, sameDiff, scalarValue| Constructor and Description |
|---|
BaseArithmeticBackpropOp() |
BaseArithmeticBackpropOp(SameDiff sameDiff,
SDVariable x,
SDVariable y,
SDVariable eps) |
| Modifier and Type | Method and Description |
|---|---|
java.util.List<long[]> |
calculateOutputShape()
Calculate
the output shape for this op
|
java.util.List<SDVariable> |
doDiff(java.util.List<SDVariable> i_v)
The actual implementation for automatic differentiation.
|
addIArgument, addIArgument, addInputArgument, addOutputArgument, addTArgument, asProperties, assertValidForExecution, builder, getDescriptor, getIArgument, getInputArgument, getOutputArgument, getTArgument, iArgs, initFromOnnx, initFromTensorFlow, inputArguments, numIArguments, numInputArguments, numOutputArguments, numTArguments, onnxName, opHash, opName, opNum, opType, outputArguments, outputVariables, outputVariables, populateInputsAndOutputsFromSameDiff, removeIArgument, removeInputArgument, removeOutputArgument, removeTArgument, sameDiffBuilder, setInputArgument, setOutputArgument, tArgs, tensorflowName, toStringarg, arg, argNames, args, attributeAdaptersForFunction, configFieldName, diff, dup, equals, f, getNumOutputs, getValue, hashCode, hasPlaceHolderInputs, isConfigProperties, larg, mappingsForFunction, onnxNames, outputVariable, outputVariablesNames, propertiesForFunction, rarg, resolvePropertiesFromSameDiffBeforeExecution, setInstanceId, setValueFor, tensorflowNamesclone, finalize, getClass, notify, notifyAll, wait, wait, waitisInplaceCallpublic BaseArithmeticBackpropOp()
public BaseArithmeticBackpropOp(SameDiff sameDiff, SDVariable x, SDVariable y, SDVariable eps)
public java.util.List<SDVariable> doDiff(java.util.List<SDVariable> i_v)
DifferentialFunctiondoDiff in class DynamicCustomOppublic java.util.List<long[]> calculateOutputShape()
DifferentialFunctioncalculateOutputShape in interface CustomOpcalculateOutputShape in class BaseDynamicTransformOp