FLOPs computation for elementwise_div op. For elementwise_div(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)
| 198 | |
| 199 | @register_flops("elementwise_div") |
| 200 | def _elementwise_div_flops(input_shapes, attrs): |
| 201 | """FLOPs computation for elementwise_div op. |
| 202 | For elementwise_div(input,other): |
| 203 | input_shapes = [shape_of_input, shape_of_other] |
| 204 | shape_of_input = [dim1, dim2, dim3 ...] |
| 205 | shape_of_other = [odim1, odim2, odim3...] |
| 206 | equation: flops = max(dim1,odim1)*max(dim2,odim2)*max()... |
| 207 | """ |
| 208 | return _elementwise_flops_compute(input_shapes, attrs) |
| 209 | |
| 210 | |
| 211 | @register_flops("gelu") |
nothing calls this directly
no test coverage detected