public class CumProdBp extends BaseReductionBp
DynamicCustomOp.DynamicCustomOpsBuilder
dimensions, keepDims
axis, bArguments, dArguments, iArguments, inplaceCall, inputArguments, outputArguments, outputVariables, tArguments
extraArgs, inPlace, sameDiff, scalarValue
Constructor and Description |
---|
CumProdBp() |
CumProdBp(INDArray origInput,
INDArray gradAtOutput,
INDArray output,
boolean exclusive,
boolean reverse,
int... axis) |
CumProdBp(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
|
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, 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 CumProdBp(SameDiff sameDiff, SDVariable origInput, SDVariable gradAtOutput, boolean exclusive, boolean reverse, int... axis)
public CumProdBp(INDArray origInput, INDArray gradAtOutput, INDArray output, boolean exclusive, boolean reverse, int... axis)
public CumProdBp()
public int getNumOutputs()
getNumOutputs
in class DifferentialFunction
protected void addArgs()
addArgs
in class BaseReductionBp
public String opName()
DynamicCustomOp
opName
in interface CustomOp
opName
in class BaseReductionBp
Copyright © 2020. All rights reserved.