(expr)
| 167 | |
| 168 | @register_functional_loader(("megengine.functional.nn", "batch_norm")) |
| 169 | def bn_func_loader(expr): |
| 170 | kwargs = expr.kwargs |
| 171 | if "compute_mode" in kwargs: |
| 172 | assert kwargs["compute_mode"] == "default" |
| 173 | kwargs.pop("compute_mode") |
| 174 | if "param_dim" in kwargs: |
| 175 | assert kwargs["param_dim"] == "dim_1c11" |
| 176 | kwargs.pop("param_dim") |
| 177 | expr.set_args_kwargs(*expr.args, **kwargs) |
| 178 | |
| 179 | |
| 180 | @register_functional_loader(("megengine.functional.math", "matmul")) |
nothing calls this directly
no test coverage detected