public abstract class BaseArithmeticBackpropOp extends BaseDynamicTransformOp
DynamicCustomOp.DynamicCustomOpsBuilder, DynamicCustomOp.SameDiffBuilder
iArguments, inplaceCall, inputArguments, outputArguments, outputVariables, tArguments
dimensions, 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, toString
arg, arg, argNames, args, attributeAdaptersForFunction, configFieldName, diff, dup, equals, f, getNumOutputs, getValue, hashCode, hasPlaceHolderInputs, isConfigProperties, larg, mappingsForFunction, onnxNames, outputVariable, outputVariablesNames, propertiesForFunction, rarg, resolvePropertiesFromSameDiffBeforeExecution, setInstanceId, setValueFor, tensorflowNames
clone, finalize, getClass, notify, notifyAll, wait, wait, wait
isInplaceCall
public BaseArithmeticBackpropOp()
public BaseArithmeticBackpropOp(SameDiff sameDiff, SDVariable x, SDVariable y, SDVariable eps)
public java.util.List<SDVariable> doDiff(java.util.List<SDVariable> i_v)
DifferentialFunction
doDiff
in class DynamicCustomOp
public java.util.List<long[]> calculateOutputShape()
DifferentialFunction
calculateOutputShape
in interface CustomOp
calculateOutputShape
in class BaseDynamicTransformOp