Skip to content
Snippets Groups Projects
Commit ba833c32 authored by Cyril Moineau's avatar Cyril Moineau
Browse files

Update meminfo default values.

parent 1c2e394a
No related branches found
No related tags found
3 merge requests!279v0.4.0,!253v0.4.0,!163Export refactor
......@@ -240,11 +240,11 @@ class ExportNode(ABC):
if "cont_size" in mem_info[idx]:
self.attributes["mem_info_cont_size"][idx] = mem_info[idx]["cont_size"]
else:
self.attributes["mem_info_cont_size"][idx] = tensor.size()
self.attributes["mem_info_cont_size"][idx] = mem_info[idx]["size"]
if "cont_offset" in mem_info[idx]:
self.attributes["mem_info_cont_offset"][idx] = mem_info[idx]["cont_offset"]
else:
self.attributes["mem_info_cont_offset"][idx] = 0
self.attributes["mem_info_cont_offset"][idx] = mem_info[idx]["offset"]
if "cont_offset" in mem_info[idx]:
self.attributes["mem_info_wrap_offset"][idx] = mem_info[idx]["wrap_offset"]
else:
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment