#include "argsParser.h"
#include "buffers.h"
#include "common.h"
#include "logger.h"
#include "parserOnnxConfig.h"
#include "NvInfer.h"
#include <cuda_runtime_api.h>
#include <cstdlib>
#include <fstream>
#include <iostream>
#include <sstream>

Include dependency graph for trt_engine_syh.hpp:

Classes
class	TensorRTEngine
	The TensorRTEngine class implements a generic TensorRT model. More...

Macros
#define	DEFINE_TRT_ENTRYPOINTS 1

#define	DEFINE_TRT_LEGACY_PARSER_ENTRYPOINT 0

Functions
samplesCommon::OnnxSampleParams	initializeSampleParams (const samplesCommon::Args &args)
	Initializes members of the params struct using the command line args.

int	main (int argc, char **argv)

Variables
const std::string	gSampleName = "TensorRT.deep_learning_AO"

Macro Definition Documentation

◆ DEFINE_TRT_ENTRYPOINTS

#define DEFINE_TRT_ENTRYPOINTS 1

TensorRTEngine.cpp This file contains the implementation of the ONNX MNIST sample. It creates the network using the MNIST onnx model. It can be run with the following command line: Command: ./sample_onnx_mnist [-h or –help] [-d=/path/to/data/dir or –datadir=/path/to/data/dir] [–useDLACore=<int>]

Definition at line 28 of file trt_engine_syh.hpp.

◆ DEFINE_TRT_LEGACY_PARSER_ENTRYPOINT

#define DEFINE_TRT_LEGACY_PARSER_ENTRYPOINT 0

Definition at line 29 of file trt_engine_syh.hpp.

Function Documentation

◆ initializeSampleParams()

samplesCommon::OnnxSampleParams initializeSampleParams ( const samplesCommon::Args & args )

Initializes members of the params struct using the command line args.

Definition at line 282 of file trt_engine_syh.hpp.

Referenced by main().

◆ main()

int main	(	int	argc,
		char **	argv
	)

Definition at line 306 of file trt_engine_syh.hpp.

References TensorRTEngine::build(), gSampleName, TensorRTEngine::infer(), and initializeSampleParams().

Variable Documentation

◆ gSampleName

const std::string gSampleName = "TensorRT.deep_learning_AO"

Definition at line 48 of file trt_engine_syh.hpp.

Referenced by main().

Classes

Macros

Functions

Variables