Revision 5d7f639769b038672750a1d83ae7c0bdd695aa9f authored by Qiao Zhang on 17 March 2022, 21:47:02 UTC, committed by jax authors on 17 March 2022, 21:51:17 UTC
name                                  cpu/op
jit_small_matmul                      2.96µs ± 2%
jit_big_matmul                        22.1µs ±21%

name                                  time/op

jit_small_matmul                      2.96µs ± 2%
jit_big_matmul                        22.7µs ±21%

PiperOrigin-RevId: 435453853
1 parent 53f52cb
History
File Mode Size
.github
benchmarks
build
cloud_tpu_colabs
docs
examples
images
jax
jaxlib
tests
third_party
.bazelrc -rw-r--r-- 16.3 KB
.bazelversion -rw-r--r-- 6 bytes
.gitignore -rw-r--r-- 291 bytes
.pre-commit-config.yaml -rw-r--r-- 563 bytes
.readthedocs.yml -rw-r--r-- 569 bytes
CHANGELOG.md -rw-r--r-- 50.5 KB
CONTRIBUTING.md -rw-r--r-- 150 bytes
LICENSE -rw-r--r-- 11.1 KB
README.md -rw-r--r-- 23.3 KB
WORKSPACE -rw-r--r-- 1.3 KB
conftest.py -rw-r--r-- 856 bytes
mypy.ini -rw-r--r-- 618 bytes
pylintrc -rw-r--r-- 1.6 KB
pytest.ini -rw-r--r-- 1.3 KB
setup.cfg -rw-r--r-- 329 bytes
setup.py -rw-r--r-- 3.4 KB

README.md

back to top