mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 12:21:27 +01:00
5da2bf945e
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/29830 as title Test Plan: na Reviewed By: hl475 Differential Revision: D18506023 fbshipit-source-id: 15693894c0aa736ab3e818bc740099f0d629cb84
16 lines
520 B
Python
16 lines
520 B
Python
from __future__ import absolute_import
|
|
from __future__ import division
|
|
from __future__ import print_function
|
|
from __future__ import unicode_literals
|
|
|
|
import operator_benchmark as op_bench
|
|
from pt import ( # noqa
|
|
add_test, batchnorm_test, cat_test, chunk_test, conv_test, # noqa
|
|
gather_test, linear_test, matmul_test, pool_test, # noqa
|
|
softmax_test, split_test, unary_test, fill_test, as_strided_test, # noqa
|
|
embeddingbag_test # noqa
|
|
)
|
|
|
|
if __name__ == "__main__":
|
|
op_bench.benchmark_runner.main()
|