()
| 6 | |
| 7 | |
| 8 | def setup_backends(): |
| 9 | if jax: |
| 10 | from jax.config import config |
| 11 | |
| 12 | config.update("jax_enable_x64", True) |
| 13 | |
| 14 | if tf: |
| 15 | from tensorflow.python.ops.numpy_ops import np_config |
| 16 | |
| 17 | np_config.enable_numpy_behavior() |
| 18 | |
| 19 | |
| 20 | def exec_bench(setup, tested_function, param_list, n_runs, warmup_runs): |
no outgoing calls
no test coverage detected