@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, positionaddress, 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, zeropublic 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.