MCPcopy Create free account
hub / github.com/XiaoMi/mace / fake_delete

Method fake_delete

tools/python/micro/mem_computer.py:111–126  ·  view source on GitHub ↗
(self, op)

Source from the content-addressed store, hash-verified

109 self.used_mem_list.append(final_mem_block)
110
111 def fake_delete(self, op):
112 for tensor_name in op.input:
113 if tensor_name in self.const_tensor_names or \
114 tensor_name in self.input_names:
115 continue
116 mace_check(tensor_name in self.ref_counts and
117 self.ref_counts[tensor_name] > 0,
118 "Invalid: ref_count is 0.")
119 self.ref_counts[tensor_name] -= 1
120 if self.ref_counts[tensor_name] is 0:
121 mem_block = self.remove_mem_block_by_name(
122 self.used_mem_list, tensor_name)
123 mace_check(mem_block is not None,
124 "error, can not find tensor: %s" % tensor_name)
125 self.free_mem_list.append(mem_block)
126 self.free_mem_list.sort(key=lambda mem_block: mem_block.size)
127
128 def fake_execute_op(self, op):
129 for i in range(len(op.output)):

Callers 1

fake_execute_opMethod · 0.95

Calls 2

mace_checkFunction · 0.90

Tested by

no test coverage detected