fix error when memory_id is VTA_MEM_ID_OUT (#4330)

diff --git a/vta/src/runtime.cc b/vta/src/runtime.cc
index 79fc0e2..b657d63 100644 (file)
--- a/vta/src/runtime.cc
+++ b/vta/src/runtime.cc
@@ -1016,7 +1016,7 @@ class CommandQueue {
           elem_bytes = VTA_ACC_ELEM_BYTES;
           break;
       case VTA_MEM_ID_OUT:
-          elem_bytes = VTA_INP_ELEM_BYTES;
+          elem_bytes = VTA_OUT_ELEM_BYTES;
           break;
       default:
           LOG(FATAL) << "Memory id not recognized:" << memory_id;