Function test_run_inference_conv_model

benchmarks/module/module_benchmark_test.py:12–20

Benchmark inference on simple convolutional network. Arguments: benchmark: Fixture that benchmarks any function passed get_clients: Fixture that returns a list of clients

(benchmark, get_clients: Callable[[int], List[Any]])

def test_run_inference_conv_model(benchmark, get_clients: Callable[[int], List[Any]]):
    """Benchmark inference on simple convolutional network.

    Arguments:
        benchmark: Fixture that benchmarks any function passed
        get_clients: Fixture that returns a list of clients
    """
    expected, mpc_model, x_mpc = set_up_model(get_clients)
    benchmark(run_inference_conv_model, expected, mpc_model, x_mpc)