org.nd4j.linalg.api.ops.impl.controlflow

Class WhileDerivative

java.lang.Object

org.nd4j.autodiff.functions.DifferentialFunction

org.nd4j.linalg.api.ops.impl.controlflow.While

org.nd4j.linalg.api.ops.impl.controlflow.WhileDerivative

All Implemented Interfaces:

CustomOp

public class WhileDerivative
extends While

While loop derivative

Author:

Adam Gibson

Field Summary

Fields inherited from class org.nd4j.linalg.api.ops.impl.controlflow.While

blockName, dummyResult, inputVars, loopBodyExecution, numLooped, outputVars, predicate, predicateExecution, targetBoolean, trueBody, trueBodyName

Fields inherited from class org.nd4j.autodiff.functions.DifferentialFunction

dimensions, extraArgs, inPlace, sameDiff, scalarValue

Constructor Summary

Constructors

Constructor and Description
WhileDerivative(While delegate)

Method Summary

Modifier and Type	Method and Description
String	getBlockName()
SDVariable[]	getInputVars()
SameDiff	getLoopBodyExecution()
int	getNumLooped()
SameDiff.SameDiffConditional	getPredicate()
SameDiff	getPredicateExecution()
SameDiff.SameDiffFunctionDefinition	getTrueBody()
String	getTrueBodyName()
String	opName() The name of the op
Op.Type	opType() The type of the op
String	tensorflowName() The opName of this function tensorflow

Methods inherited from class org.nd4j.linalg.api.ops.impl.controlflow.While

addIArgument, addInputArgument, addOutputArgument, addTArgument, assertValidForExecution, calculateOutputShape, doDiff, getDescriptor, getIArgument, getInputArgument, getOutputArgument, getTArgument, iArgs, incrementLoopCounter, initFromOnnx, initFromTensorFlow, inputArguments, isInplaceCall, numIArguments, numInputArguments, numOutputArguments, numTArguments, onnxName, opHash, outputArguments, outputVariables, populateInputsAndOutputsFromSameDiff, removeIArgument, removeInputArgument, removeOutputArgument, removeTArgument, tArgs, tensorflowNames, toString

Methods inherited from class org.nd4j.autodiff.functions.DifferentialFunction

arg, args, asProperties, attributeAdaptersForFunction, configFieldName, diff, dup, equals, f, getValue, hashCode, hasPlaceHolderInputs, isConfigProperties, larg, mappingsForFunction, onnxNames, opNum, outputVariables, propertiesForFunction, rarg, resolvePropertiesFromSameDiffBeforeExecution, setInstanceId, setValueFor

Methods inherited from class java.lang.Object

clone, finalize, getClass, notify, notifyAll, wait, wait, wait

Constructor Detail

WhileDerivative

public WhileDerivative(While delegate)

Method Detail

getTrueBody

public SameDiff.SameDiffFunctionDefinition getTrueBody()

getTrueBodyName

public String getTrueBodyName()

getPredicate

public SameDiff.SameDiffConditional getPredicate()

getPredicateExecution

public SameDiff getPredicateExecution()

getInputVars

public SDVariable[] getInputVars()

getBlockName

public String getBlockName()

getLoopBodyExecution

public SameDiff getLoopBodyExecution()

getNumLooped

public int getNumLooped()

opName

public String opName()

Description copied from class: DifferentialFunction

The name of the op

Specified by:

opName in interface CustomOp

Overrides:

opName in class While

Returns:

opType

public Op.Type opType()

Description copied from class: DifferentialFunction

The type of the op

Overrides:

opType in class While

Returns:

tensorflowName

public String tensorflowName()

Description copied from class: DifferentialFunction

The opName of this function tensorflow

Overrides:

tensorflowName in class While

Returns: