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

Class If

java.lang.Object

org.nd4j.autodiff.functions.DifferentialFunction

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

All Implemented Interfaces:

CustomOp

Direct Known Subclasses:

IfDerivative

public class If
extends DifferentialFunction
implements CustomOp

Equivalent to tensorflow's conditional op. Runs one of 2 SameDiff.SameDiffFunctionDefinition depending on a predicate SameDiff.SameDiffConditional

Author:

Adam Gibson

Field Summary

Fields

Modifier and Type	Field and Description
protected String	blockName
protected SDVariable	dummyResult
protected SameDiff.SameDiffFunctionDefinition	falseBody
protected SameDiff	falseBodyExecution
protected String	falseBodyName
protected SDVariable[]	inputVars
protected SameDiff	loopBodyExecution
protected SDVariable[]	outputVars
protected SameDiff.SameDiffConditional	predicate
protected SameDiff	predicateExecution
protected SDVariable	targetBoolean
protected SameDiff.SameDiffFunctionDefinition	trueBody
protected Boolean	trueBodyExecuted
protected String	trueBodyName

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

dimensions, extraArgs, inPlace, sameDiff, scalarValue

Constructor Summary

Constructors

Constructor and Description
If(If ifStatement)
If(String blockName, SameDiff parent, SDVariable[] inputVars, SameDiff.SameDiffFunctionDefinition conditionBody, SameDiff.SameDiffConditional predicate, SameDiff.SameDiffFunctionDefinition trueBody, SameDiff.SameDiffFunctionDefinition falseBody)

Method Summary

Modifier and Type	Method and Description
void	addIArgument(int... arg)
void	addInputArgument(INDArray... arg)
void	addOutputArgument(INDArray... arg)
void	addTArgument(double... arg)
void	assertValidForExecution() Asserts a valid state for execution, otherwise throws an ND4JIllegalStateException
List<int[]>	calculateOutputShape() Calculate the output shape for this op
List<SDVariable>	doDiff(List<SDVariable> f1) The actual implementation for automatic differentiation.
void	exectedTrueOrFalse(boolean trueBodyExecuted) Toggle whether the true body was executed or the false body
CustomOpDescriptor	getDescriptor() Get the custom op descriptor if one is available.
Integer	getIArgument(int index)
INDArray	getInputArgument(int index)
INDArray	getOutputArgument(int index)
Double	getTArgument(int index)
int[]	iArgs()
void	initFromOnnx(OnnxProto3.NodeProto node, SameDiff initWith, Map<String,OnnxProto3.AttributeProto> attributesForNode, OnnxProto3.GraphProto graph) Iniitialize the function from the given OnnxProto3.NodeProto
void	initFromTensorFlow(NodeDef nodeDef, SameDiff initWith, Map<String,AttrValue> attributesForNode, GraphDef graph) Initialize the function from the given NodeDef
INDArray[]	inputArguments()
boolean	isInplaceCall() This method returns true if op is supposed to be executed inplace
int	numIArguments()
int	numInputArguments()
int	numOutputArguments()
int	numTArguments()
String	onnxName() The opName of this function in onnx
long	opHash() This method returns LongHash of the opName()
String	opName() The name of the op
Op.Type	opType() The type of the op
INDArray[]	outputArguments()
SDVariable[]	outputVariables(String baseName) Return the output functions for this differential function.
void	populateInputsAndOutputsFromSameDiff() Attempt to populate inputs and outputs from samediff.
void	removeIArgument(Integer arg)
void	removeInputArgument(INDArray arg)
void	removeOutputArgument(INDArray arg)
void	removeTArgument(Double arg)
double[]	tArgs()
String	tensorflowName() The opName of this function tensorflow
String	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, tensorflowNames

Methods inherited from class java.lang.Object

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

Field Detail

loopBodyExecution

protected SameDiff loopBodyExecution

predicateExecution

protected SameDiff predicateExecution

falseBodyExecution

protected SameDiff falseBodyExecution

predicate

protected SameDiff.SameDiffConditional predicate

trueBody

protected SameDiff.SameDiffFunctionDefinition trueBody

falseBody

protected SameDiff.SameDiffFunctionDefinition falseBody

blockName

protected String blockName

trueBodyName

protected String trueBodyName

falseBodyName

protected String falseBodyName

inputVars

protected SDVariable[] inputVars

trueBodyExecuted

protected Boolean trueBodyExecuted

targetBoolean

protected SDVariable targetBoolean

dummyResult

protected SDVariable dummyResult

outputVars

protected SDVariable[] outputVars

Constructor Detail

If

public If(If ifStatement)

If

public If(String blockName,
          SameDiff parent,
          SDVariable[] inputVars,
          SameDiff.SameDiffFunctionDefinition conditionBody,
          SameDiff.SameDiffConditional predicate,
          SameDiff.SameDiffFunctionDefinition trueBody,
          SameDiff.SameDiffFunctionDefinition falseBody)

Method Detail

exectedTrueOrFalse

public void exectedTrueOrFalse(boolean trueBodyExecuted)

Toggle whether the true body was executed or the false body

Parameters:

trueBodyExecuted -

outputVariables

public SDVariable[] outputVariables(String baseName)

Description copied from class: DifferentialFunction

Return the output functions for this differential function.

Specified by:

outputVariables in class DifferentialFunction

Returns:

doDiff

public List<SDVariable> doDiff(List<SDVariable> f1)

Description copied from class: DifferentialFunction

The actual implementation for automatic differentiation.

Specified by:

doDiff in class DifferentialFunction

Returns:

toString

public String toString()

Overrides:

toString in class Object

opName

public String opName()

Description copied from class: DifferentialFunction

The name of the op

Specified by:

opName in interface CustomOp

Overrides:

opName in class DifferentialFunction

Returns:

opHash

public long opHash()

Description copied from interface: CustomOp

This method returns LongHash of the opName()

Specified by:

opHash in interface CustomOp

Returns:

isInplaceCall

public boolean isInplaceCall()

Description copied from interface: CustomOp

This method returns true if op is supposed to be executed inplace

Specified by:

isInplaceCall in interface CustomOp

Returns:

outputArguments

public INDArray[] outputArguments()

Specified by:

outputArguments in interface CustomOp

inputArguments

public INDArray[] inputArguments()

Specified by:

inputArguments in interface CustomOp

iArgs

public int[] iArgs()

Specified by:

iArgs in interface CustomOp

tArgs

public double[] tArgs()

Specified by:

tArgs in interface CustomOp

addIArgument

public void addIArgument(int... arg)

Specified by:

addIArgument in interface CustomOp

removeIArgument

public void removeIArgument(Integer arg)

Specified by:

removeIArgument in interface CustomOp

getIArgument

public Integer getIArgument(int index)

Specified by:

getIArgument in interface CustomOp

numIArguments

public int numIArguments()

Specified by:

numIArguments in interface CustomOp

addTArgument

public void addTArgument(double... arg)

Specified by:

addTArgument in interface CustomOp

removeTArgument

public void removeTArgument(Double arg)

Specified by:

removeTArgument in interface CustomOp

getTArgument

public Double getTArgument(int index)

Specified by:

getTArgument in interface CustomOp

numTArguments

public int numTArguments()

Specified by:

numTArguments in interface CustomOp

addInputArgument

public void addInputArgument(INDArray... arg)

Specified by:

addInputArgument in interface CustomOp

removeInputArgument

public void removeInputArgument(INDArray arg)

Specified by:

removeInputArgument in interface CustomOp

getInputArgument

public INDArray getInputArgument(int index)

Specified by:

getInputArgument in interface CustomOp

numInputArguments

public int numInputArguments()

Specified by:

numInputArguments in interface CustomOp

addOutputArgument

public void addOutputArgument(INDArray... arg)

Specified by:

addOutputArgument in interface CustomOp

removeOutputArgument

public void removeOutputArgument(INDArray arg)

Specified by:

removeOutputArgument in interface CustomOp

getOutputArgument

public INDArray getOutputArgument(int index)

Specified by:

getOutputArgument in interface CustomOp

numOutputArguments

public int numOutputArguments()

Specified by:

numOutputArguments in interface CustomOp

opType

public Op.Type opType()

Description copied from class: DifferentialFunction

The type of the op

Overrides:

opType in class DifferentialFunction

Returns:

initFromTensorFlow

public void initFromTensorFlow(NodeDef nodeDef,
                               SameDiff initWith,
                               Map<String,AttrValue> attributesForNode,
                               GraphDef graph)

Description copied from class: DifferentialFunction

Initialize the function from the given NodeDef

Specified by:

initFromTensorFlow in class DifferentialFunction

initFromOnnx

public void initFromOnnx(OnnxProto3.NodeProto node,
                         SameDiff initWith,
                         Map<String,OnnxProto3.AttributeProto> attributesForNode,
                         OnnxProto3.GraphProto graph)

Description copied from class: DifferentialFunction

Iniitialize the function from the given OnnxProto3.NodeProto

Specified by:

initFromOnnx in class DifferentialFunction

calculateOutputShape

public List<int[]> calculateOutputShape()

Description copied from class: DifferentialFunction

Calculate the output shape for this op

Specified by:

calculateOutputShape in interface CustomOp

Overrides:

calculateOutputShape in class DifferentialFunction

Returns:

getDescriptor

public CustomOpDescriptor getDescriptor()

Description copied from interface: CustomOp

Get the custom op descriptor if one is available.

Specified by:

getDescriptor in interface CustomOp

Returns:

assertValidForExecution

public void assertValidForExecution()

Description copied from interface: CustomOp

Asserts a valid state for execution, otherwise throws an ND4JIllegalStateException

Specified by:

assertValidForExecution in interface CustomOp

populateInputsAndOutputsFromSameDiff

public void populateInputsAndOutputsFromSameDiff()

Description copied from interface: CustomOp

Attempt to populate inputs and outputs from samediff. This method will not attempt to fill in integer or double aruguments.

Specified by:

populateInputsAndOutputsFromSameDiff in interface CustomOp

onnxName

public String onnxName()

Description copied from class: DifferentialFunction

The opName of this function in onnx

Specified by:

onnxName in class DifferentialFunction

Returns:

tensorflowName

public String tensorflowName()

Description copied from class: DifferentialFunction

The opName of this function tensorflow

Specified by:

tensorflowName in class DifferentialFunction

Returns: