TensorRT
8.0.3
|
Public Member Functions | |
virtual bool | execute (int32_t batchSize, void *const *bindings) noexcept=0 |
virtual bool | enqueue (int32_t batchSize, void *const *bindings, cudaStream_t stream, cudaEvent_t *inputConsumed) noexcept=0 |
virtual void | setDebugSync (bool sync) noexcept=0 |
virtual bool | getDebugSync () const noexcept=0 |
virtual void | setProfiler (IProfiler *) noexcept=0 |
virtual IProfiler * | getProfiler () const noexcept=0 |
virtual const ICudaEngine & | getEngine () const noexcept=0 |
virtual void | setName (const char *name) noexcept=0 |
virtual const char * | getName () const noexcept=0 |
virtual void | setDeviceMemory (void *memory) noexcept=0 |
virtual Dims | getStrides (int32_t bindingIndex) const noexcept=0 |
virtual bool | setOptimizationProfile (int32_t profileIndex) noexcept=0 |
virtual int32_t | getOptimizationProfile () const noexcept=0 |
virtual bool | setBindingDimensions (int32_t bindingIndex, Dims dimensions) noexcept=0 |
virtual Dims | getBindingDimensions (int32_t bindingIndex) const noexcept=0 |
virtual bool | setInputShapeBinding (int32_t bindingIndex, int32_t const *data) noexcept=0 |
virtual bool | getShapeBinding (int32_t bindingIndex, int32_t *data) const noexcept=0 |
virtual bool | allInputDimensionsSpecified () const noexcept=0 |
virtual bool | allInputShapesSpecified () const noexcept=0 |
virtual void | setErrorRecorder (IErrorRecorder *recorder) noexcept=0 |
virtual IErrorRecorder * | getErrorRecorder () const noexcept=0 |
virtual bool | executeV2 (void *const *bindings) noexcept=0 |
virtual bool | enqueueV2 (void *const *bindings, cudaStream_t stream, cudaEvent_t *inputConsumed) noexcept=0 |
virtual bool | setOptimizationProfileAsync (int32_t profileIndex, cudaStream_t stream) noexcept=0 |
Copyright © 2024 NVIDIA Corporation
Privacy Policy |
Manage My Privacy |
Do Not Sell or Share My Data |
Terms of Service |
Accessibility |
Corporate Policies |
Product Security |
Contact