org.bytedeco.tensorrt.nvinfer

Class IElementWiseLayer

java.lang.Object

org.bytedeco.javacpp.Pointer

org.bytedeco.tensorrt.nvinfer.INoCopy

org.bytedeco.tensorrt.nvinfer.ILayer

org.bytedeco.tensorrt.nvinfer.IElementWiseLayer

All Implemented Interfaces:

AutoCloseable

@Namespace(value="nvinfer1")
 @NoOffset
 @Properties(inherit=nvinfer.class)
public class IElementWiseLayer
extends ILayer

\class IElementWiseLayer \brief A elementwise layer in a network definition. This layer applies a per-element binary operation between corresponding elements of two tensors. The input tensors must have the same rank. For each dimension, their lengths must match, or one of them must be one. In the latter case, the tensor is broadcast along that axis. The output tensor has the same rank as the inputs. For each output dimension, its length is equal to the lengths of the corresponding input dimensions if they match, otherwise it is equal to the length that is not one. \warning When running this layer on the DLA with Int8 data type, the dynamic ranges of two input tensors shall be equal. If the dynamic ranges are generated using calibrator, the largest value shall be used. \warning Do not inherit from this class, as doing so will break forward-compatibility of the API and ABI.

Nested Class Summary

Nested classes/interfaces inherited from class org.bytedeco.javacpp.Pointer

Pointer.CustomDeallocator, Pointer.Deallocator, Pointer.NativeDeallocator, Pointer.ReferenceCounter

Field Summary

Fields inherited from class org.bytedeco.javacpp.Pointer

address, capacity, limit, position

Constructor Summary

Constructors

Constructor and Description
IElementWiseLayer(Pointer p) Pointer cast constructor.

Method Summary

Modifier and Type	Method and Description
nvinfer.ElementWiseOperation	getOperation() \brief Get the binary operation for the layer.
void	setOperation(int op)
void	setOperation(nvinfer.ElementWiseOperation op) \brief Set the binary operation for the layer.

Methods inherited from class org.bytedeco.tensorrt.nvinfer.ILayer

getInput, getMetadata, getName, getNbInputs, getNbOutputs, getOutput, getOutputType, getPrecision, getType, outputTypeIsSet, precisionIsSet, resetOutputType, resetPrecision, setInput, setMetadata, setMetadata, setName, setName, setOutputType, setOutputType, setPrecision, setPrecision

Methods inherited from class org.bytedeco.javacpp.Pointer

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

Methods inherited from class java.lang.Object

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

Constructor Detail

IElementWiseLayer

public IElementWiseLayer(Pointer p)

Pointer cast constructor. Invokes Pointer(Pointer).

Method Detail

setOperation

@NoException(value=true)
public void setOperation(nvinfer.ElementWiseOperation op)

\brief Set the binary operation for the layer. DLA supports only kSUM, kPROD, kMAX, kMIN, and kSUB.

See Also:

getOperation(), ElementWiseOperation, getBiasWeights()

setOperation

@NoException(value=true)
public void setOperation(@Cast(value="nvinfer1::ElementWiseOperation")
                                                  int op)

getOperation

@NoException(value=true)
public nvinfer.ElementWiseOperation getOperation()

\brief Get the binary operation for the layer.

See Also:

setOperation(), ElementWiseOperation, setBiasWeights()