public class NormMaxBp extends BaseReductionBp
DynamicCustomOp.DynamicCustomOpsBuilder
dimensions, keepDims
axis, bArguments, dArguments, iArguments, inplaceCall, inputArguments, outputArguments, outputVariables, tArguments
extraArgs, inPlace, ownName, ownNameSetWithDefault, sameDiff, scalarValue
Constructor and Description |
---|
NormMaxBp() |
NormMaxBp(INDArray origInput,
INDArray gradAtOutput,
INDArray output,
boolean keepDims,
int... dimensions) |
NormMaxBp(SameDiff sameDiff,
SDVariable origInput,
SDVariable gradAtOutput,
boolean keepDims,
int... dimensions) |
Modifier and Type | Method and Description |
---|---|
String |
opName()
This method returns op opName as string
|
addArgs, calculateOutputDataTypes
addBArgument, addDArgument, addIArgument, addIArgument, addInputArgument, addOutputArgument, addTArgument, assertValidForExecution, bArgs, builder, calculateOutputShape, calculateOutputShape, clearArrays, dArgs, doDiff, getBArgument, getDescriptor, getIArgument, getInputArgument, getOutputArgument, getTArgument, iArgs, initFromOnnx, initFromTensorFlow, inputArguments, numBArguments, numDArguments, numIArguments, numInputArguments, numOutputArguments, numTArguments, onnxName, opHash, opNum, opType, outputArguments, outputVariables, outputVariables, removeIArgument, removeInputArgument, removeOutputArgument, removeTArgument, setInputArgument, setInputArguments, setOutputArgument, tArgs, tensorflowName, toString, wrapFilterNull, wrapOrNull, wrapOrNull
arg, arg, argNames, args, attributeAdaptersForFunction, configFieldName, diff, dup, equals, getNumOutputs, getValue, hashCode, isConfigProperties, larg, mappingsForFunction, onnxNames, outputs, outputVariable, outputVariablesNames, propertiesForFunction, rarg, replaceArg, setInstanceId, setPropertiesForFunction, setValueFor, tensorflowNames
clone, finalize, getClass, notify, notifyAll, wait, wait, wait
isInplaceCall
public NormMaxBp(SameDiff sameDiff, SDVariable origInput, SDVariable gradAtOutput, boolean keepDims, int... dimensions)
public NormMaxBp(INDArray origInput, INDArray gradAtOutput, INDArray output, boolean keepDims, int... dimensions)
public NormMaxBp()
public String opName()
DynamicCustomOp
opName
in interface CustomOp
opName
in class BaseReductionBp
Copyright © 2021. All rights reserved.