TensorRT
8.0.2
|
Public Member Functions | |
virtual void | setMaxBatchSize (int32_t batchSize) noexcept=0 |
virtual int32_t | getMaxBatchSize () const noexcept=0 |
virtual bool | platformHasFastFp16 () const noexcept=0 |
virtual bool | platformHasFastInt8 () const noexcept=0 |
virtual int32_t | getMaxDLABatchSize () const noexcept=0 |
virtual int32_t | getNbDLACores () const noexcept=0 |
virtual void | setGpuAllocator (IGpuAllocator *allocator) noexcept=0 |
virtual nvinfer1::IBuilderConfig * | createBuilderConfig () noexcept=0 |
virtual nvinfer1::ICudaEngine * | buildEngineWithConfig (INetworkDefinition &network, IBuilderConfig &config) noexcept=0 |
virtual nvinfer1::INetworkDefinition * | createNetworkV2 (NetworkDefinitionCreationFlags flags) noexcept=0 |
virtual nvinfer1::IOptimizationProfile * | createOptimizationProfile () noexcept=0 |
virtual void | setErrorRecorder (IErrorRecorder *recorder) noexcept=0 |
virtual IErrorRecorder * | getErrorRecorder () const noexcept=0 |
virtual void | reset () noexcept=0 |
virtual bool | platformHasTf32 () const noexcept=0 |
virtual nvinfer1::IHostMemory * | buildSerializedNetwork (INetworkDefinition &network, IBuilderConfig &config) noexcept=0 |
virtual bool | isNetworkSupported (INetworkDefinition const &network, IBuilderConfig const &config) const noexcept=0 |