#include <NvInfer.h>

Inheritance diagram for nvinfer1::IIfConditional:

Public Member Functions
IConditionLayer *	setCondition (ITensor &condition) noexcept
	Set the condition tensor for this If-Conditional construct. More...

IIfConditionalOutputLayer *	addOutput (ITensor &trueSubgraphOutput, ITensor &falseSubgraphOutput) noexcept
	Add an If-conditional output. More...

IIfConditionalInputLayer *	addInput (ITensor &input) noexcept
	Add an If-conditional input. More...

void	setName (const char *name) noexcept
	Set the name of the conditional. More...

const char *	getName () const noexcept
	Return the name of the conditional. More...

Protected Attributes
apiv::VIfConditional *	mImpl

Additional Inherited Members
Protected Member Functions inherited from nvinfer1::INoCopy
	INoCopy (const INoCopy &other)=delete

INoCopy &	operator= (const INoCopy &other)=delete

	INoCopy (INoCopy &&other)=delete

INoCopy &	operator= (INoCopy &&other)=delete

Detailed Description

Helper for constructing conditionally-executed subgraphs.

An If-conditional conditionally executes part of the network according to the following pseudo-code:

If condition is true then: output = trueSubgraph(trueInputs); Else output = falseSubgraph(falseInputs); Emit output

Condition is a 0D boolean tensor (representing a scalar). trueSubgraph represents a network subgraph that is executed when condition is evaluated to True. falseSubgraph represents a network subgraph that is executed when condition is evaluated to False.

The following constraints apply to If-conditionals:

Both the trueSubgraph and falseSubgraph must be defined.
The number of output tensors in both subgraphs is the same.
The type and shape of each output tensor from true/false subgraphs are the same.

Member Function Documentation

◆ addInput()

IIfConditionalInputLayer * nvinfer1::IIfConditional::addInput ( ITensor & input )

inlinenoexcept

Add an If-conditional input.

Parameters

input An input to the conditional that can be used by either or both of the conditional’s subgraphs.

See also: IIfConditionalInputLayer

◆ addOutput()

IIfConditionalOutputLayer * nvinfer1::IIfConditional::addOutput	(	ITensor &	trueSubgraphOutput,
		ITensor &	falseSubgraphOutput
	)

inlinenoexcept

Add an If-conditional output.

Parameters

trueSubgraphOutput	The output of the subgraph executed when the conditional evaluates to true.
falseSubgraphOutput	The output of the subgraph executed when the conditional evaluates to false.

Each output layer of an IIfConditional represents a single output of either the true-subgraph or the false-subgraph of an IIfConditional, depending on which subgraph was executed.