summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
author김용섭/On-Device Lab(SR)/Engineer/삼성전자 <yons.kim@samsung.com>2019-04-19 08:38:44 (GMT)
committer오형석/On-Device Lab(SR)/Staff Engineer/삼성전자 <hseok82.oh@samsung.com>2019-04-19 08:38:44 (GMT)
commitec76f7f9ec5842623df5d4f144b81af308ffe15e (patch)
tree97190adefa1f8dc9733eddb256ae252c31229eab
parentc5403134115c204720dc8bee846811b00e3fc9f7 (diff)
downloadnnfw-ec76f7f9ec5842623df5d4f144b81af308ffe15e.zip
nnfw-ec76f7f9ec5842623df5d4f144b81af308ffe15e.tar.gz
nnfw-ec76f7f9ec5842623df5d4f144b81af308ffe15e.tar.bz2
Add printing info for filled memory in model_parser (#5028)
* Add printing info for filled memory in model_parser Add printing info for filled memory such as weight in model_parser Signed-off-by: Yongseop Kim <yons.kim@samsung.com> * fix failed format-check
-rw-r--r--tools/tflitefile_tool/model_printer.py7
1 files changed, 5 insertions, 2 deletions
diff --git a/tools/tflitefile_tool/model_printer.py b/tools/tflitefile_tool/model_printer.py
index ad06fa7..fda0291 100644
--- a/tools/tflitefile_tool/model_printer.py
+++ b/tools/tflitefile_tool/model_printer.py
@@ -134,9 +134,12 @@ class ModelPrinter(object):
def PrintTotalMemory(self):
total_memory = 0
+ filled_memory = 0 # only memory for constant
for tensor in self.op_parser.GetAllTensors():
+ if tensor.tf_buffer.DataLength() != 0:
+ filled_memory += tensor.memory_size
total_memory += tensor.memory_size
from tensor_printer import ConvertBytesToHuman
- print("Expected total memory for allocating all tensors: {0}".format(
- ConvertBytesToHuman(total_memory)))
+ print("Expected TOTAL memory: {0}".format(ConvertBytesToHuman(total_memory)))
+ print("Expected FILLED memory: {0}".format(ConvertBytesToHuman(filled_memory)))