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

Method scratch_size_of_squeeze

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

Source from the content-addressed store, hash-verified

182 return block_size * 4 * k_channels * float_bytes
183
184 def scratch_size_of_squeeze(self, op_def):
185 input0_dims = self.get_op_input_dims(op_def, 0)
186 return len(input0_dims) * self.get_data_bytes(mace_pb2.DT_INT32) * 2
187
188 def scratch_size_eltwise(self, op_def):
189 input0_dims = self.get_op_input_dims(op_def, 0)

Callers

nothing calls this directly

Calls 2

get_op_input_dimsMethod · 0.95
get_data_bytesMethod · 0.95

Tested by

no test coverage detected