FAST/Algorithms/NeuralNetwork/InferenceEngines/TensorRTEngine.cpp file

Contents

Namespaces

namespace fast
FAST namespace - All FAST code lives in this namespace.

Classes

class fast::Logger
struct fast::Destroy

Functions

static auto getTensorShape(nvinfer1::Dims dims) -> TensorShape
auto elementSize(nvinfer1::DataType t) -> unsigned int
auto safeCudaMalloc(size_t memSize) -> void*
auto shapeToDims(TensorShape shape) -> nvinfer1::Dims

Variables

fast::Logger gLogger

Defines

#define CUDA_CHECK(status)