FLOPs computation for elementwise_mul op. For elementwise_mul(input,other): input_shapes = [shape_of_input, shape_of_other] shape_of_input = [dim1, dim2, dim3 ...] shape_of_other = [odim1, odim2, odim3...] equation: flops = max(dim1, odim1) * max(dim2, odim2)* max
(input_shapes, attrs)
| 186 | |
| 187 | @register_flops("elementwise_mul") |
| 188 | def _elementwise_mul_flops(input_shapes, attrs): |
| 189 | """FLOPs computation for elementwise_mul op. |
| 190 | For elementwise_mul(input,other): |
| 191 | input_shapes = [shape_of_input, shape_of_other] |
| 192 | shape_of_input = [dim1, dim2, dim3 ...] |
| 193 | shape_of_other = [odim1, odim2, odim3...] |
| 194 | equation: flops = max(dim1, odim1) * max(dim2, odim2)* max()... |
| 195 | """ |
| 196 | return _elementwise_flops_compute(input_shapes, attrs) |
| 197 | |
| 198 | |
| 199 | @register_flops("elementwise_div") |
nothing calls this directly
no test coverage detected