(op)
| 145 | |
| 146 | |
| 147 | def count_linear(op): |
| 148 | total_mul = op.inputs("Y")[0].shape()[0] |
| 149 | numel = np.prod(op.outputs("Out")[0].shape()[1:]) |
| 150 | total_ops = total_mul * numel |
| 151 | total_ops = abs(total_ops) |
| 152 | return total_ops |
| 153 | |
| 154 | |
| 155 | def count_pool2d(op): |
no test coverage detected