#include "NvInferSafeMemAllocator.h"
#include "NvInferSafePlugin.h"
#include "NvInferSafeRecorder.h"
#include <algorithm>
#include <cuda_fp16.h>
#include <iterator>
Go to the source code of this file.
|
| bool | nvinfer2::safe::operator== (PhysicalDims const &d0, PhysicalDims const &d1) noexcept |
| |
| bool | nvinfer2::safe::operator!= (PhysicalDims const &d0, PhysicalDims const &d1) noexcept |
| |
| ErrorCode | nvinfer2::safe::createTRTGraph (ITRTGraph *&graph, void const *buffer, int64_t bufferSize, ISafeRecorder &recorder, bool trtManagedScratch=true, ISafeMemAllocator *allocator=nullptr) noexcept |
| | The C factory function which serves as an entry point to TRT that creates an instance of a ITRTGraph class. It will performs the following operations: More...
|
| |
| ErrorCode | nvinfer2::safe::createTRTGraphWithSo (ITRTGraph *&graph, void const *buffer, int64_t bufferSize, AsciiChar const *companionSoPath, ISafeRecorder &recorder, bool trtManagedScratch=true, ISafeMemAllocator *allocator=nullptr) noexcept |
| | createTRTGraph() variant that additionally binds the engine's companion shared object. More...
|
| |
| ErrorCode | nvinfer2::safe::destroyTRTGraph (ITRTGraph *&graph) noexcept |
| | Toplevel graph destructor. More...
|
| |