tinyengine
tinyengine copied to clipboard
some api used in "GeneralMemoryScheduler.py"
Excuse me, I cannot trace the exact definition of the following api which are related to op
layermem["MAC"] = op.get_macs() layermem["activation"] = op.get_activation_size() layermem["scale"] = op.get_scale_size() layermem["runtime"] = op.get_sbuf_size() layermem["kernel"] = op.get_kbuf_size()
Where can we find the definition of get_macs()/get_activation_size()/get_scale_size()/get_sbuf_size()/get_kbuf_size() ? Could you please help comment that? Thanks
Hi @waquey, thanks for your interest in our work! Please refer to the implementation of operators in code_generator/operators. The abstraction of these APIs is in code_generator/operators/basic_utils.py.
Close due to inactivity. Feel free to reopen.