tensorRT serialization OK
This commit is contained in:
+1
-1
@@ -25,7 +25,7 @@ public:
|
||||
dnnType *output;
|
||||
cudaStream_t stream;
|
||||
|
||||
NetworkRT(Network *net);
|
||||
NetworkRT(Network *net, const char *name);
|
||||
virtual ~NetworkRT();
|
||||
|
||||
/**
|
||||
|
||||
Reference in New Issue
Block a user