better rt inference

This commit is contained in:
Francesco Gatti
2017-08-21 12:10:17 +00:00
parent c63ac6b590
commit 0aa9de4ce8
3 changed files with 22 additions and 8 deletions
+4
View File
@@ -139,6 +139,10 @@ dnnType* NetworkRT::infer(dataDim_t &dim, dnnType* data) {
return output;
}
void NetworkRT::enqueue() {
contextRT->enqueue(1, buffersRT, stream, nullptr);
}
ILayer* NetworkRT::convert_layer(ITensor *input, Layer *l) {
layerType_t type = l->getLayerType();