Revision 115ddcbe73af08da7e7c5e6f414bc647f2f620d4 authored by John QiangZhang on 02 August 2022, 17:44:59 UTC, committed by jax authors on 09 August 2022, 21:30:06 UTC
Quick update: * clean the function_alias parameters. * disable custom_gradient and run through jax test. PiperOrigin-RevId: 464830257
1 parent 6d9512a
File | Mode | Size |
---|---|---|
benchmarks | ||
filecheck | ||
notebooks | ||
third_party | ||
BUILD | -rw-r--r-- | 16.9 KB |
ann_test.py | -rw-r--r-- | 8.6 KB |
api_test.py | -rw-r--r-- | 281.6 KB |
api_util_test.py | -rw-r--r-- | 2.4 KB |
array_interoperability_test.py | -rw-r--r-- | 9.6 KB |
array_test.py | -rw-r--r-- | 7.5 KB |
batching_test.py | -rw-r--r-- | 51.9 KB |
callback_test.py | -rw-r--r-- | 7.1 KB |
checkify_test.py | -rw-r--r-- | 25.8 KB |
clear_backends_test.py | -rw-r--r-- | 1.3 KB |
compilation_cache_test.py | -rw-r--r-- | 14.1 KB |
core_test.py | -rw-r--r-- | 23.4 KB |
custom_linear_solve_test.py | -rw-r--r-- | 14.9 KB |
custom_object_test.py | -rw-r--r-- | 11.7 KB |
custom_root_test.py | -rw-r--r-- | 8.1 KB |
debug_nans_test.py | -rw-r--r-- | 7.0 KB |
debugger_test.py | -rw-r--r-- | 12.1 KB |
debugging_primitives_test.py | -rw-r--r-- | 24.0 KB |
distributed_multinode_test.py | -rw-r--r-- | 2.3 KB |
distributed_test.py | -rw-r--r-- | 3.6 KB |
dtypes_test.py | -rw-r--r-- | 23.1 KB |
dynamic_api_test.py | -rw-r--r-- | 41.5 KB |
errors_test.py | -rw-r--r-- | 11.3 KB |
fft_test.py | -rw-r--r-- | 16.5 KB |
generated_fun_test.py | -rw-r--r-- | 8.0 KB |
gfile_cache_test.py | -rw-r--r-- | 2.8 KB |
global_device_array_test.py | -rw-r--r-- | 15.4 KB |
heap_profiler_test.py | -rw-r--r-- | 1.2 KB |
host_callback_test.py | -rw-r--r-- | 107.0 KB |
host_callback_to_tf_test.py | -rw-r--r-- | 7.7 KB |
image_test.py | -rw-r--r-- | 17.5 KB |
infeed_test.py | -rw-r--r-- | 3.9 KB |
jax_jit_test.py | -rw-r--r-- | 7.8 KB |
jax_to_ir_test.py | -rw-r--r-- | 5.4 KB |
jaxpr_effects_test.py | -rw-r--r-- | 27.8 KB |
jaxpr_util_test.py | -rw-r--r-- | 3.2 KB |
jet_test.py | -rw-r--r-- | 16.1 KB |
lax_autodiff_test.py | -rw-r--r-- | 51.9 KB |
lax_control_flow_test.py | -rw-r--r-- | 91.1 KB |
lax_numpy_einsum_test.py | -rw-r--r-- | 8.8 KB |
lax_numpy_indexing_test.py | -rw-r--r-- | 59.6 KB |
lax_numpy_test.py | -rw-r--r-- | 294.4 KB |
lax_numpy_vectorize_test.py | -rw-r--r-- | 7.4 KB |
lax_scipy_sparse_test.py | -rw-r--r-- | 18.1 KB |
lax_scipy_test.py | -rw-r--r-- | 25.1 KB |
lax_test.py | -rw-r--r-- | 144.2 KB |
lax_vmap_test.py | -rw-r--r-- | 36.9 KB |
linalg_test.py | -rw-r--r-- | 69.8 KB |
lobpcg_test.py | -rw-r--r-- | 14.0 KB |
mesh_utils_test.py | -rw-r--r-- | 10.9 KB |
metadata_test.py | -rw-r--r-- | 4.0 KB |
multi_device_test.py | -rw-r--r-- | 8.2 KB |
multibackend_test.py | -rw-r--r-- | 7.1 KB |
name_stack_test.py | -rw-r--r-- | 20.1 KB |
nn_test.py | -rw-r--r-- | 10.1 KB |
ode_test.py | -rw-r--r-- | 8.1 KB |
optimizers_test.py | -rw-r--r-- | 10.1 KB |
pickle_test.py | -rw-r--r-- | 3.5 KB |
pjit_test.py | -rw-r--r-- | 89.1 KB |
pmap_test.py | -rw-r--r-- | 108.8 KB |
polynomial_test.py | -rw-r--r-- | 4.9 KB |
profiler_test.py | -rw-r--r-- | 7.8 KB |
python_callback_test.py | -rw-r--r-- | 12.5 KB |
qdwh_test.py | -rw-r--r-- | 9.3 KB |
random_test.py | -rw-r--r-- | 69.9 KB |
remote_transfer_test.py | -rw-r--r-- | 1.6 KB |
scipy_fft_test.py | -rw-r--r-- | 3.2 KB |
scipy_interpolate_test.py | -rw-r--r-- | 2.6 KB |
scipy_ndimage_test.py | -rw-r--r-- | 6.1 KB |
scipy_optimize_test.py | -rw-r--r-- | 6.9 KB |
scipy_signal_test.py | -rw-r--r-- | 18.5 KB |
scipy_stats_test.py | -rw-r--r-- | 33.4 KB |
sparse_test.py | -rw-r--r-- | 102.8 KB |
sparsify_test.py | -rw-r--r-- | 19.3 KB |
stack_test.py | -rw-r--r-- | 1.5 KB |
state_test.py | -rw-r--r-- | 14.6 KB |
stax_test.py | -rw-r--r-- | 10.8 KB |
svd_test.py | -rw-r--r-- | 8.8 KB |
transfer_guard_test.py | -rw-r--r-- | 8.3 KB |
tree_util_test.py | -rw-r--r-- | 20.1 KB |
util_test.py | -rw-r--r-- | 2.9 KB |
version_test.py | -rw-r--r-- | 1.8 KB |
x64_context_test.py | -rw-r--r-- | 4.8 KB |
xla_bridge_test.py | -rw-r--r-- | 8.5 KB |
xla_interpreter_test.py | -rw-r--r-- | 1.1 KB |
xmap_test.py | -rw-r--r-- | 76.9 KB |
Computing file changes ...