TNN icon indicating copy to clipboard operation
TNN copied to clipboard

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

Open DamonsJ opened this issue 1 year ago • 0 comments

希望能在profile中加入算子调用次数及算子调用的kernel名称 加入算子调用次数: profile.cc line 180 for (auto p : profiling_data_) { 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; } }

DamonsJ avatar Mar 02 '23 03:03 DamonsJ