DynamicCustomOp.DynamicCustomOpsBuilder, DynamicCustomOp.SameDiffBuilder
inplaceCall, outputVariables
dimensions, extraArgs, inPlace, sameDiff, scalarValue
Constructor and Description |
---|
BatchNormDerivative() |
BatchNormDerivative(SameDiff sameDiff,
SDVariable[] inputFunctions,
INDArray[] inputArrays,
INDArray[] outputArrays,
boolean inPlace,
boolean applyGamma,
boolean applyBeta,
double epsilon) |
Modifier and Type | Method and Description |
---|---|
List<SDVariable> |
doDiff(List<SDVariable> f1)
The actual implementation for automatic differentiation.
|
String |
opName()
This method returns op opName as string
|
addArgs, initFromOnnx, initFromTensorFlow, mappingsForFunction, onnxName, propertiesForFunction, tensorflowName
addIArgument, addInputArgument, addOutputArgument, addTArgument, asProperties, assertValidForExecution, builder, calculateOutputShape, getDescriptor, getIArgument, getInputArgument, getOutputArgument, getTArgument, iArgs, inputArguments, numIArguments, numInputArguments, numOutputArguments, numTArguments, opHash, opNum, opType, outputArguments, outputVariables, outputVariables, populateInputsAndOutputsFromSameDiff, removeIArgument, removeInputArgument, removeOutputArgument, removeTArgument, sameDiffBuilder, setInputArgument, setOutputArgument, tArgs, toString, updateInputsFromSameDiff
arg, args, attributeAdaptersForFunction, configFieldName, diff, dup, equals, f, getValue, hashCode, hasPlaceHolderInputs, isConfigProperties, larg, onnxNames, rarg, resolvePropertiesFromSameDiffBeforeExecution, setInstanceId, setValueFor, tensorflowNames
clone, finalize, getClass, notify, notifyAll, wait, wait, wait
isInplaceCall
public BatchNormDerivative(SameDiff sameDiff, SDVariable[] inputFunctions, INDArray[] inputArrays, INDArray[] outputArrays, boolean inPlace, boolean applyGamma, boolean applyBeta, double epsilon)
public BatchNormDerivative()
public String opName()
DynamicCustomOp
public List<SDVariable> doDiff(List<SDVariable> f1)
DifferentialFunction
Copyright © 2018. All rights reserved.