@Namespace(value="nvinfer1") @NoOffset @Properties(inherit=nvinfer.class) public class IIfConditional extends INoCopy
Pointer.CustomDeallocator, Pointer.Deallocator, Pointer.NativeDeallocator, Pointer.ReferenceCounter
Constructor and Description |
---|
IIfConditional(Pointer p)
Pointer cast constructor.
|
Modifier and Type | Method and Description |
---|---|
IIfConditionalInputLayer |
addInput(ITensor input)
\brief Add an If-conditional input.
|
IIfConditionalOutputLayer |
addOutput(ITensor trueSubgraphOutput,
ITensor falseSubgraphOutput)
\brief Add an If-conditional output.
|
String |
getName()
\brief Return the name of the conditional.
|
IConditionLayer |
setCondition(ITensor condition)
\brief Set the condition tensor for this If-Conditional construct.
|
void |
setName(BytePointer name) |
void |
setName(String name)
\brief Set the name of the conditional.
|
address, asBuffer, asByteBuffer, availablePhysicalBytes, calloc, capacity, capacity, close, deallocate, deallocate, deallocateReferences, deallocator, deallocator, equals, fill, formatBytes, free, getDirectBufferAddress, getPointer, getPointer, getPointer, getPointer, hashCode, interruptDeallocatorThread, isNull, isNull, limit, limit, malloc, maxBytes, maxPhysicalBytes, memchr, memcmp, memcpy, memmove, memset, offsetAddress, offsetof, offsetof, parseBytes, physicalBytes, physicalBytesInaccurate, position, position, put, realloc, referenceCount, releaseReference, retainReference, setNull, sizeof, sizeof, toString, totalBytes, totalCount, totalPhysicalBytes, withDeallocator, zero
public IIfConditional(Pointer p)
Pointer(Pointer)
.@NoException(value=true) public IConditionLayer setCondition(@ByRef ITensor condition)
condition
- The condition tensor that will determine which subgraph to execute.
\p condition tensor must be a 0D execution tensor (scalar) with type DataType::kBOOL.IConditionLayer
@NoException(value=true) public IIfConditionalOutputLayer addOutput(@ByRef ITensor trueSubgraphOutput, @ByRef ITensor falseSubgraphOutput)
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.IIfConditionalOutputLayer
@NoException(value=true) public IIfConditionalInputLayer addInput(@ByRef ITensor input)
input
- An input to the conditional that can be used by either or both of the conditionalâs subgraphs.IIfConditionalInputLayer
@NoException(value=true) public void setName(String name)
getName()
@NoException(value=true) public void setName(@Cast(value="const char*") BytePointer name)
@NoException(value=true) public String getName()
setName()
Copyright © 2024. All rights reserved.