better rt inference

2017-08-21 12:10:17 +00:00
parent c63ac6b590
commit 0aa9de4ce8
3 changed files with 22 additions and 8 deletions
@@ -32,6 +32,7 @@ public:
        Do inferece
    */
    dnnType* infer(dataDim_t &dim, dnnType* data);
+    void enqueue();    

    nvinfer1::ILayer* convert_layer(nvinfer1::ITensor *input, Layer *l);
    nvinfer1::ILayer* convert_layer(nvinfer1::ITensor *input, Conv2d *l);
@@ -62,4 +63,4 @@ template<typename T> T readBUF(const char*& buffer)
 }

 }
-#endif //NETWORKRT_H
+#endif //NETWORKRT_H