Tencent / TNN

TNN: developed by Tencent Youtu Lab and Guangying Lab, a uniform deep learning inference framework for mobile、desktop and server. TNN is distinguished by several outstanding features, including its cross-platform capability, high performance, model compression and code pruning. Based on ncnn and Rapidnet, TNN further strengthens the support and performance optimization for mobile devices, and also draws on the advantages of good extensibility and high performance from existed open source efforts. TNN has been deployed in multiple Apps from Tencent, such as Mobile QQ, Weishi, Pitu, etc. Contributions are welcome to work in collaborative with us and make TNN a better framework.
Other
4.4k stars 770 forks source link

profile中加入算子调用次数及算子调用的kernel名称 #1894

Open DamonsJ opened 1 year ago

DamonsJ commented 1 year ago

希望能在profile中加入算子调用次数及算子调用的kernel名称 加入算子调用次数: profile.cc line 180 for (auto p : profilingdata) { if (summary_map.find(p->op_name) != summary_map.end()) { if (do_average) summary_map[p->op_name][0] += p->kernel_time / p->count; else summary_map[p->op_name][0] += p->kernel_time; summary_map[p->op_name][1] = summary_map[p->op_name][1] + 1; } }