pytorch

1
from pt import (  # noqa: F401  # noqa: F401
2
    add_test,
3
    ao_sparsifier_test,
4
    as_strided_test,
5
    batchnorm_test,
6
    binary_test,
7
    cat_test,
8
    channel_shuffle_test,
9
    chunk_test,
10
    conv_test,
11
    diag_test,
12
    embeddingbag_test,
13
    fill_test,
14
    gather_test,
15
    groupnorm_test,
16
    hardsigmoid_test,
17
    hardswish_test,
18
    instancenorm_test,
19
    interpolate_test,
20
    layernorm_test,
21
    linear_test,
22
    matmul_test,
23
    nan_to_num_test,
24
    pool_test,
25
    remainder_test,
26
    softmax_test,
27
    split_test,
28
    sum_test,
29
    tensor_to_test,
30
)
31

32
import operator_benchmark as op_bench
33

34
if __name__ == "__main__":
35
    op_bench.benchmark_runner.main()
36