trt_engine.hpp File Reference

#include "buffers.hpp"
#include "logger.h"
#include "parserOnnxConfig.hpp"
#include "util.hpp"
#include "/opt/MagAOX/vendor/TensorRT-10.0.0.6/include/NvInfer.h"
#include <cuda_runtime_api.h>
#include <cstdlib>
#include <fstream>
#include <iostream>
#include <sstream>

Include dependency graph for trt_engine.hpp:

Go to the source code of this file.

Classes
class	TensorrtEngine

Macros
#define	DEFINE_TRT_ENTRYPOINTS   1
#define	DEFINE_TRT_LEGACY_PARSER_ENTRYPOINT   0

Typedefs
using	preciseStopwatch = Stopwatch<>

Variables
Logger	gLogger

Macro Definition Documentation

DEFINE_TRT_ENTRYPOINTS

#define DEFINE_TRT_ENTRYPOINTS   1

Definition at line 2 of file trt_engine.hpp.

DEFINE_TRT_LEGACY_PARSER_ENTRYPOINT

#define DEFINE_TRT_LEGACY_PARSER_ENTRYPOINT   0

Definition at line 3 of file trt_engine.hpp.

Typedef Documentation

preciseStopwatch

using preciseStopwatch = Stopwatch<>

Definition at line 19 of file trt_engine.hpp.

Variable Documentation

gLogger

Logger gLogger

Definition at line 20 of file trt_engine.hpp.

Referenced by TensorrtEngine::build(), and TensorrtEngine::load().