Skip to content
Snippets Groups Projects
Commit c977243b authored by Nathan Thoumine's avatar Nathan Thoumine
Browse files

add general graph class description

parent 91dbf3e2
No related branches found
No related tags found
No related merge requests found
......@@ -32,6 +32,15 @@ typedef struct
unsigned int nIO;
} IOGraphDesc;
/**
* @class Graph
* @brief Manages the lifecycle and execution of a neural network graph using TensorRT.
*
* The Graph class encapsulates the functionality required to manage, configure, and execute
* a neural network graph for inference using NVIDIA's TensorRT. This includes loading models
* from ONNX or TensorRT files, setting the CUDA device and data types, managing calibration
* for INT8 precision, and running inference in both synchronous and asynchronous modes.
*/
class Graph
{
public:
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment