public class CumSumBp extends BaseReductionBp
DynamicCustomOp.DynamicCustomOpsBuilderdimensions, keepDimsaxis, bArguments, dArguments, iArguments, inplaceCall, inputArguments, outputArguments, outputVariables, tArgumentsextraArgs, inPlace, ownName, ownNameSetWithDefault, sameDiff, scalarValue| Constructor and Description |
|---|
CumSumBp() |
CumSumBp(INDArray origInput,
INDArray gradAtOutput,
INDArray output,
boolean exclusive,
boolean reverse,
int... axis) |
CumSumBp(SameDiff sameDiff,
SDVariable origInput,
SDVariable gradAtOutput,
boolean exclusive,
boolean reverse,
int... axis) |
| Modifier and Type | Method and Description |
|---|---|
protected void |
addArgs() |
int |
getNumOutputs() |
String |
opName()
This method returns op opName as string
|
calculateOutputDataTypesaddBArgument, 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, wrapOrNullarg, arg, argNames, args, attributeAdaptersForFunction, configFieldName, diff, dup, equals, getValue, hashCode, isConfigProperties, larg, mappingsForFunction, onnxNames, outputs, outputVariable, outputVariablesNames, propertiesForFunction, rarg, replaceArg, setInstanceId, setPropertiesForFunction, setValueFor, tensorflowNamesclone, finalize, getClass, notify, notifyAll, wait, wait, waitisInplaceCallpublic CumSumBp(SameDiff sameDiff, SDVariable origInput, SDVariable gradAtOutput, boolean exclusive, boolean reverse, int... axis)
public CumSumBp(INDArray origInput, INDArray gradAtOutput, INDArray output, boolean exclusive, boolean reverse, int... axis)
public CumSumBp()
public int getNumOutputs()
getNumOutputs in class DifferentialFunctionprotected void addArgs()
addArgs in class BaseReductionBppublic String opName()
DynamicCustomOpopName in interface CustomOpopName in class BaseReductionBpCopyright © 2021. All rights reserved.