Google Git
Sign in
android / platform / external / pytorch / 9b147961c4 / . / benchmarks / operator_benchmark
tree: d50f358814ce25c87ef6c2d4deb29a6d40b57722 [path history] [tgz]
  1. ops/
  2. __init__.py
  3. batchnorm_benchmark.py
  4. benchmark_caffe2.py
  5. benchmark_core.py
  6. benchmark_pytorch.py
  7. benchmark_runner.py
  8. benchmark_test_generator.py
  9. benchmark_utils.py
  10. README.md
benchmarks/operator_benchmark/README.md

PyTorch/Caffe2 Operator Micro-benchmarks

Run benchmarks

Go to pytorch-src-folder/benchmarks python -m operator_benchmark.ops.matmul_test

should report the execution time of matmul operator with PyTorch and Caffe2