| 1115 | } |
| 1116 | |
| 1117 | struct tensor_comparer { |
| 1118 | bool operator()(const tensor_statistics & a, const tensor_statistics & b) const { |
| 1119 | std::string layer, name_a, name_b; |
| 1120 | ; |
| 1121 | process_tensor_name(a.tensor, layer, name_a); |
| 1122 | process_tensor_name(b.tensor, layer, name_b); |
| 1123 | return name_a < name_b || (name_a == name_b && a.total_sqract > b.total_sqract); |
| 1124 | } |
| 1125 | }; |
| 1126 | std::sort(ts.begin(), ts.end(), tensor_comparer()); |
| 1127 | |
| 1128 | struct weighted_stats { |