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

Method scratch_size_of_concat

tools/python/micro/scratch_computer.py:192–194  ·  view source on GitHub ↗
(self, op_def)

Source from the content-addressed store, hash-verified

190 return len(input0_dims) * self.get_data_bytes(mace_pb2.DT_INT32) * 3
191
192 def scratch_size_of_concat(self, op_def):
193 # On a 64bit operating system, one pointer data need 8 bytes
194 return len(op_def.input) * self.get_data_bytes(mace_pb2.DT_INT32) * 3
195
196 def scratch_size_of_reshape(self, op_def):
197 shape_dims = self.get_op_input_dims(op_def, 1)

Callers

nothing calls this directly

Calls 1

get_data_bytesMethod · 0.95

Tested by

no test coverage detected