Revision a002643a4aff1803c578b3520464367e7e9e3bac authored by Peter Hawkins on 02 March 2023, 16:23:50 UTC, committed by jax authors on 02 March 2023, 16:24:30 UTC
Work around pytype bug. It seems that the line from functools import cached_property causes pytype to give up on the entire module. Avoid the member import to fix the type inference. PiperOrigin-RevId: 513544106
1 parent a9421a8
File | Mode | Size |
---|---|---|
cpu | ||
cuda | ||
gpu | ||
mlir | ||
rocm | ||
BUILD | -rw-r--r-- | 3.9 KB |
README.md | -rw-r--r-- | 402 bytes |
cpu_feature_guard.c | -rw-r--r-- | 5.2 KB |
ducc_fft.py | -rw-r--r-- | 4.5 KB |
gpu_linalg.py | -rw-r--r-- | 2.4 KB |
gpu_prng.py | -rw-r--r-- | 2.7 KB |
gpu_rnn.py | -rw-r--r-- | 4.7 KB |
gpu_solver.py | -rw-r--r-- | 16.4 KB |
gpu_sparse.py | -rw-r--r-- | 12.1 KB |
handle_pool.h | -rw-r--r-- | 3.3 KB |
hlo_helpers.py | -rw-r--r-- | 2.8 KB |
init.py | -rw-r--r-- | 629 bytes |
jax.bzl | -rw-r--r-- | 7.7 KB |
kernel_helpers.h | -rw-r--r-- | 1.7 KB |
kernel_pybind11_helpers.h | -rw-r--r-- | 1.6 KB |
lapack.py | -rw-r--r-- | 23.1 KB |
setup.cfg | -rw-r--r-- | 38 bytes |
setup.py | -rw-r--r-- | 2.4 KB |
symlink_files.bzl | -rw-r--r-- | 6.0 KB |
![swh spinner](/static/img/swh-spinner.gif)
Computing file changes ...