public abstract class BaseArithmeticBackpropOp extends BaseDynamicTransformOp
DynamicCustomOp.DynamicCustomOpsBuilder, DynamicCustomOp.SameDiffBuilder
inplaceCall, outputVariables
dimensions, extraArgs, inPlace, sameDiff, scalarValue
Constructor and Description |
---|
BaseArithmeticBackpropOp() |
BaseArithmeticBackpropOp(SameDiff sameDiff,
SDVariable x,
SDVariable y,
SDVariable eps) |
Modifier and Type | Method and Description |
---|---|
List<int[]> |
calculateOutputShape()
Calculate
the output shape for this op
|
List<SDVariable> |
doDiff(List<SDVariable> i_v)
The actual implementation for automatic differentiation.
|
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, updateInputsFromSameDiff
arg, args, attributeAdaptersForFunction, configFieldName, diff, dup, equals, f, getValue, hashCode, hasPlaceHolderInputs, isConfigProperties, larg, mappingsForFunction, onnxNames, 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 List<SDVariable> doDiff(List<SDVariable> i_v)
DifferentialFunction
doDiff
in class DynamicCustomOp
public List<int[]> calculateOutputShape()
DifferentialFunction
calculateOutputShape
in interface CustomOp
calculateOutputShape
in class BaseDynamicTransformOp
Copyright © 2018. All rights reserved.