@Namespace(value="nvinfer1::apiv") @Properties(inherit=nvinfer.class) public class VNetworkDefinition extends VRoot
Pointer.CustomDeallocator, Pointer.Deallocator, Pointer.NativeDeallocator, Pointer.ReferenceCounter
Constructor and Description |
---|
VNetworkDefinition(Pointer p)
Pointer cast constructor.
|
getPointer, position
address, asBuffer, asByteBuffer, availablePhysicalBytes, calloc, capacity, capacity, close, deallocate, deallocate, deallocateReferences, deallocator, deallocator, equals, fill, formatBytes, free, getDirectBufferAddress, getPointer, getPointer, getPointer, hashCode, interruptDeallocatorThread, isNull, isNull, limit, limit, malloc, maxBytes, maxPhysicalBytes, memchr, memcmp, memcpy, memmove, memset, offsetAddress, offsetof, offsetof, parseBytes, physicalBytes, physicalBytesInaccurate, position, put, realloc, referenceCount, releaseReference, retainReference, setNull, sizeof, sizeof, toString, totalBytes, totalCount, totalPhysicalBytes, withDeallocator, zero
public VNetworkDefinition(Pointer p)
Pointer(Pointer)
.@NoException(value=true) public ITensor addInput(String name, nvinfer.DataType type, @ByVal @Cast(value="nvinfer1::Dims*") Dims32 dimensions)
@NoException(value=true) public ITensor addInput(@Cast(value="const char*") BytePointer name, @Cast(value="nvinfer1::DataType") int type, @ByVal @Cast(value="nvinfer1::Dims*") Dims32 dimensions)
@NoException(value=true) public void markOutput(@ByRef ITensor tensor)
@NoException(value=true) public IConvolutionLayer addConvolution(@ByRef ITensor input, int nbOutputMaps, @ByVal DimsHW kernelSize, @ByVal Weights kernelWeights, @ByVal Weights biasWeights)
@NoException(value=true) public IFullyConnectedLayer addFullyConnected(@ByRef ITensor input, int nbOutputs, @ByVal Weights kernelWeights, @ByVal Weights biasWeights)
@NoException(value=true) public IActivationLayer addActivation(@ByRef ITensor input, nvinfer.ActivationType type)
@NoException(value=true) public IActivationLayer addActivation(@ByRef ITensor input, @Cast(value="nvinfer1::ActivationType") int type)
@NoException(value=true) public IPoolingLayer addPooling(@ByRef ITensor input, nvinfer.PoolingType type, @ByVal DimsHW windowSize)
@NoException(value=true) public IPoolingLayer addPooling(@ByRef ITensor input, @Cast(value="nvinfer1::PoolingType") int type, @ByVal DimsHW windowSize)
@NoException(value=true) public ILRNLayer addLRN(@ByRef ITensor input, int window, float alpha, float beta, float k)
@NoException(value=true) public IScaleLayer addScale(@ByRef ITensor input, nvinfer.ScaleMode mode, @ByVal Weights shift, @ByVal Weights scale, @ByVal Weights power)
@NoException(value=true) public IScaleLayer addScale(@ByRef ITensor input, @Cast(value="nvinfer1::ScaleMode") int mode, @ByVal Weights shift, @ByVal Weights scale, @ByVal Weights power)
@NoException(value=true) public ISoftMaxLayer addSoftMax(@ByRef ITensor input)
@NoException(value=true) public IConcatenationLayer addConcatenation(@Cast(value="nvinfer1::ITensor*const*") PointerPointer inputs, int nbInputs)
@NoException(value=true) public IConcatenationLayer addConcatenation(@ByPtrPtr ITensor inputs, int nbInputs)
@NoException(value=true) public IDeconvolutionLayer addDeconvolution(@ByRef ITensor input, int nbOutputMaps, @ByVal DimsHW kernelSize, @ByVal Weights kernelWeights, @ByVal Weights biasWeights)
@NoException(value=true) public IElementWiseLayer addElementWise(@ByRef ITensor input1, @ByRef ITensor input2, nvinfer.ElementWiseOperation op)
@NoException(value=true) public IElementWiseLayer addElementWise(@ByRef ITensor input1, @ByRef ITensor input2, @Cast(value="nvinfer1::ElementWiseOperation") int op)
@NoException(value=true) public IUnaryLayer addUnary(@ByRef ITensor input, nvinfer.UnaryOperation operation)
@NoException(value=true) public IUnaryLayer addUnary(@ByRef ITensor input, @Cast(value="nvinfer1::UnaryOperation") int operation)
@NoException(value=true) public IPaddingLayer addPadding(@ByRef ITensor input, @ByVal DimsHW prePadding, @ByVal DimsHW postPadding)
@NoException(value=true) public IShuffleLayer addShuffle(@ByRef ITensor input)
@NoException(value=true) public int getNbLayers()
@NoException(value=true) public ILayer getLayer(int index)
@NoException(value=true) public int getNbInputs()
@NoException(value=true) public ITensor getInput(int index)
@NoException(value=true) public int getNbOutputs()
@NoException(value=true) public ITensor getOutput(int index)
@NoException(value=true) public IReduceLayer addReduce(@ByRef ITensor input, nvinfer.ReduceOperation operation, @Cast(value="uint32_t") int reduceAxes, @Cast(value="bool") boolean keepDimensions)
@NoException(value=true) public IReduceLayer addReduce(@ByRef ITensor input, @Cast(value="nvinfer1::ReduceOperation") int operation, @Cast(value="uint32_t") int reduceAxes, @Cast(value="bool") boolean keepDimensions)
@NoException(value=true) public ITopKLayer addTopK(@ByRef ITensor input, nvinfer.TopKOperation op, int k, @Cast(value="uint32_t") int reduceAxes)
@NoException(value=true) public ITopKLayer addTopK(@ByRef ITensor input, @Cast(value="nvinfer1::TopKOperation") int op, int k, @Cast(value="uint32_t") int reduceAxes)
@NoException(value=true) public IGatherLayer addGather(@ByRef ITensor data, @ByRef ITensor indices, int axis)
@NoException(value=true) public IRaggedSoftMaxLayer addRaggedSoftMax(@ByRef ITensor input, @ByRef ITensor bounds)
@NoException(value=true) public IMatrixMultiplyLayer addMatrixMultiply(@ByRef ITensor input0, nvinfer.MatrixOperation op0, @ByRef ITensor input1, nvinfer.MatrixOperation op1)
@NoException(value=true) public IMatrixMultiplyLayer addMatrixMultiply(@ByRef ITensor input0, @Cast(value="nvinfer1::MatrixOperation") int op0, @ByRef ITensor input1, @Cast(value="nvinfer1::MatrixOperation") int op1)
@NoException(value=true) public IConstantLayer addConstant(@ByVal @Cast(value="nvinfer1::Dims*") Dims32 dimensions, @ByVal Weights weights)
@NoException(value=true) public IRNNv2Layer addRNNv2(@ByRef ITensor input, int layerCount, int hiddenSize, int maxSeqLen, nvinfer.RNNOperation op)
@NoException(value=true) public IRNNv2Layer addRNNv2(@ByRef ITensor input, int layerCount, int hiddenSize, int maxSeqLen, @Cast(value="nvinfer1::RNNOperation") int op)
@NoException(value=true) public IIdentityLayer addIdentity(@ByRef ITensor input)
@NoException(value=true) public void removeTensor(@ByRef ITensor tensor)
@NoException(value=true) public void unmarkOutput(@ByRef ITensor tensor)
@NoException(value=true) public IPluginV2Layer addPluginV2(@Cast(value="nvinfer1::ITensor*const*") PointerPointer inputs, int nbInputs, @ByRef IPluginV2 plugin)
@NoException(value=true) public IPluginV2Layer addPluginV2(@ByPtrPtr ITensor inputs, int nbInputs, @ByRef IPluginV2 plugin)
@NoException(value=true) public ISliceLayer addSlice(@ByRef ITensor input, @ByVal @Cast(value="nvinfer1::Dims*") Dims32 start, @ByVal @Cast(value="nvinfer1::Dims*") Dims32 size, @ByVal @Cast(value="nvinfer1::Dims*") Dims32 stride)
@NoException(value=true) public void setName(String name)
@NoException(value=true) public void setName(@Cast(value="const char*") BytePointer name)
@NoException(value=true) public String getName()
@NoException(value=true) public IShapeLayer addShape(@ByRef ITensor input)
@Cast(value="bool") @NoException(value=true) public boolean hasImplicitBatchDimension()
@Cast(value="bool") @NoException(value=true) public boolean markOutputForShapes(@ByRef ITensor tensor)
@Cast(value="bool") @NoException(value=true) public boolean unmarkOutputForShapes(@ByRef ITensor tensor)
@NoException(value=true) public IParametricReLULayer addParametricReLU(@ByRef ITensor input, @ByRef ITensor slope)
@NoException(value=true) public IConvolutionLayer addConvolutionNd(@ByRef ITensor input, int nbOutputMaps, @ByVal @Cast(value="nvinfer1::Dims*") Dims32 kernelSize, @ByVal Weights kernelWeights, @ByVal Weights biasWeights)
@NoException(value=true) public IPoolingLayer addPoolingNd(@ByRef ITensor input, nvinfer.PoolingType type, @ByVal @Cast(value="nvinfer1::Dims*") Dims32 windowSize)
@NoException(value=true) public IPoolingLayer addPoolingNd(@ByRef ITensor input, @Cast(value="nvinfer1::PoolingType") int type, @ByVal @Cast(value="nvinfer1::Dims*") Dims32 windowSize)
@NoException(value=true) public IDeconvolutionLayer addDeconvolutionNd(@ByRef ITensor input, int nbOutputMaps, @ByVal @Cast(value="nvinfer1::Dims*") Dims32 kernelSize, @ByVal Weights kernelWeights, @ByVal Weights biasWeights)
@NoException(value=true) public IScaleLayer addScaleNd(@ByRef ITensor input, nvinfer.ScaleMode mode, @ByVal Weights shift, @ByVal Weights scale, @ByVal Weights power, int channelAxis)
@NoException(value=true) public IScaleLayer addScaleNd(@ByRef ITensor input, @Cast(value="nvinfer1::ScaleMode") int mode, @ByVal Weights shift, @ByVal Weights scale, @ByVal Weights power, int channelAxis)
@NoException(value=true) public IResizeLayer addResize(@ByRef ITensor input)
@Cast(value="bool") @NoException(value=true) public boolean hasExplicitPrecision()
@NoException(value=true) public ILoop addLoop()
@NoException(value=true) public ISelectLayer addSelect(@ByRef ITensor condition, @ByRef ITensor thenInput, @ByRef ITensor elseInput)
@NoException(value=true) public IFillLayer addFill(@ByVal @Cast(value="nvinfer1::Dims*") Dims32 dimensions, nvinfer.FillOperation op)
@NoException(value=true) public IFillLayer addFill(@ByVal @Cast(value="nvinfer1::Dims*") Dims32 dimensions, @Cast(value="nvinfer1::FillOperation") int op)
@NoException(value=true) public IPaddingLayer addPaddingNd(@ByRef ITensor input, @ByVal @Cast(value="nvinfer1::Dims*") Dims32 prePadding, @ByVal @Cast(value="nvinfer1::Dims*") Dims32 postPadding)
@Cast(value="bool") @NoException(value=true) public boolean setWeightsName(@ByVal Weights weights, String name)
@Cast(value="bool") @NoException(value=true) public boolean setWeightsName(@ByVal Weights weights, @Cast(value="const char*") BytePointer name)
@NoException(value=true) public void setErrorRecorder(IErrorRecorder recorder)
@NoException(value=true) public IErrorRecorder getErrorRecorder()
@NoException(value=true) public IDequantizeLayer addDequantize(@ByRef ITensor input, @ByRef ITensor scale)
@NoException(value=true) public IQuantizeLayer addQuantize(@ByRef ITensor input, @ByRef ITensor scale)
@NoException(value=true) public IGatherLayer addGatherV2(@ByRef ITensor data, @ByRef ITensor indices, nvinfer.GatherMode mode)
@NoException(value=true) public IGatherLayer addGatherV2(@ByRef ITensor data, @ByRef ITensor indices, @Cast(value="nvinfer1::GatherMode") int mode)
@NoException(value=true) public IIfConditional addIfConditional()
@NoException(value=true) public IScatterLayer addScatter(@ByRef ITensor data, @ByRef ITensor indices, @ByRef ITensor updates, nvinfer.ScatterMode mode)
@NoException(value=true) public IScatterLayer addScatter(@ByRef ITensor data, @ByRef ITensor indices, @ByRef ITensor updates, @Cast(value="nvinfer1::ScatterMode") int mode)
@NoException(value=true) public IEinsumLayer addEinsum(@Cast(value="nvinfer1::ITensor*const*") PointerPointer inputs, int nbInputs, String equation)
@NoException(value=true) public IEinsumLayer addEinsum(@ByPtrPtr ITensor inputs, int nbInputs, String equation)
@NoException(value=true) public IEinsumLayer addEinsum(@ByPtrPtr ITensor inputs, int nbInputs, @Cast(value="const char*") BytePointer equation)
@NoException(value=true) public IAssertionLayer addAssertion(@ByRef ITensor condition, String message)
@NoException(value=true) public IAssertionLayer addAssertion(@ByRef ITensor condition, @Cast(value="const char*") BytePointer message)
@NoException(value=true) public IOneHotLayer addOneHot(@ByRef ITensor indices, @ByRef ITensor values, @ByRef ITensor depth, int axis)
@NoException(value=true) public INonZeroLayer addNonZero(@ByRef ITensor input)
@NoException(value=true) public IGridSampleLayer addGridSample(@ByRef ITensor input, @ByRef ITensor grid)
@NoException(value=true) public INMSLayer addNMS(@ByRef ITensor boxes, @ByRef ITensor scores, @ByRef ITensor maxOutputBoxesPerClass)
@NoException(value=true) public IReverseSequenceLayer addReverseSequence(@ByRef ITensor input, @ByRef ITensor sequenceLens)
@NoException(value=true) public INormalizationLayer addNormalization(@ByRef ITensor input, @ByRef ITensor scale, @ByRef ITensor bias, @Cast(value="uint32_t") int axesMask)
@NoException(value=true) public ICastLayer addCast(@ByRef ITensor input, nvinfer.DataType toType)
@NoException(value=true) public ICastLayer addCast(@ByRef ITensor input, @Cast(value="nvinfer1::DataType") int toType)
@ByRef @NoException(value=true) public IBuilder getBuilder()
Copyright © 2024. All rights reserved.