Compute flops for Rsqrt operation.
(graph, node)
| 82 | |
| 83 | @ops.RegisterStatistics("Rsqrt", "flops") |
| 84 | def _rsqrt_flops(graph, node): |
| 85 | """Compute flops for Rsqrt operation.""" |
| 86 | # Rsqrt(x) = 1 / sqrt(x) |
| 87 | return _unary_op_flops(graph, node, ops_per_element=2) |
| 88 | |
| 89 | |
| 90 | @ops.RegisterStatistics("Log", "flops") |
nothing calls this directly
no test coverage detected