Flatten batch to be checked

2020-04-23 01:18:27 +02:00
parent 1cbf18c21e
commit 2a6f7db4af
2 changed files with 11 additions and 6 deletions
@@ -46,7 +46,8 @@ int main(int argc, char *argv[]) {
        TIMER_STOP

        // control output
-        for(int o=1; o<netRT.getBuffersN(); o++) {
+        std::cout<<"Output Buffers: "<<netRT.getBuffersN()-1<<"\n";
+	for(int o=1; o<netRT.getBuffersN(); o++) {
            for(int b=1; b<BATCH_SIZE; b++) {
                dnnType *out_d = (dnnType*) netRT.buffersRT[o];
                dnnType *out0_d = out_d;