This commit is contained in:
Marcos Luciano
2023-06-05 16:07:37 -03:00
parent 66a6754b77
commit 79d4a0a8cd
4 changed files with 9 additions and 10 deletions

View File

@@ -41,6 +41,12 @@
#include "layers/pooling_layer.h"
#include "layers/reorg_layer.h"
#if NV_TENSORRT_MAJOR >= 8
#define INT int32_t
#else
#define INT int
#endif
struct NetworkInfo
{
std::string inputBlobName;