Revision 2c23c947a52331f6e4352bffc9bb185bc1c8cacd authored by jax authors on 16 March 2022, 19:55:19 UTC, committed by jax authors on 16 March 2022, 19:55:19 UTC
PiperOrigin-RevId: 435131902
| File | Mode | Size |
|---|---|---|
| benchmarks | ||
| filecheck | ||
| notebooks | ||
| third_party | ||
| ann_test.py | -rw-r--r-- | 6.9 KB |
| api_test.py | -rw-r--r-- | 243.4 KB |
| api_util_test.py | -rw-r--r-- | 2.4 KB |
| array_interoperability_test.py | -rw-r--r-- | 8.2 KB |
| batching_test.py | -rw-r--r-- | 51.5 KB |
| callback_test.py | -rw-r--r-- | 7.1 KB |
| checkify_test.py | -rw-r--r-- | 19.9 KB |
| compilation_cache_test.py | -rw-r--r-- | 14.1 KB |
| core_test.py | -rw-r--r-- | 19.2 KB |
| custom_linear_solve_test.py | -rw-r--r-- | 14.1 KB |
| custom_object_test.py | -rw-r--r-- | 12.5 KB |
| custom_root_test.py | -rw-r--r-- | 8.1 KB |
| debug_nans_test.py | -rw-r--r-- | 6.5 KB |
| djax_test.py | -rw-r--r-- | 4.8 KB |
| dtypes_test.py | -rw-r--r-- | 18.6 KB |
| errors_test.py | -rw-r--r-- | 11.3 KB |
| fft_test.py | -rw-r--r-- | 16.1 KB |
| file_system_cache_test.py | -rw-r--r-- | 5.5 KB |
| generated_fun_test.py | -rw-r--r-- | 8.0 KB |
| global_device_array_test.py | -rw-r--r-- | 14.7 KB |
| heap_profiler_test.py | -rw-r--r-- | 1.2 KB |
| host_callback_test.py | -rw-r--r-- | 108.8 KB |
| host_callback_to_tf_test.py | -rw-r--r-- | 7.7 KB |
| image_test.py | -rw-r--r-- | 16.6 KB |
| infeed_test.py | -rw-r--r-- | 3.9 KB |
| jax_jit_test.py | -rw-r--r-- | 7.7 KB |
| jax_to_ir_test.py | -rw-r--r-- | 5.2 KB |
| jaxpr_util_test.py | -rw-r--r-- | 3.2 KB |
| jet_test.py | -rw-r--r-- | 15.6 KB |
| lax_autodiff_test.py | -rw-r--r-- | 50.8 KB |
| lax_control_flow_test.py | -rw-r--r-- | 76.4 KB |
| lax_numpy_einsum_test.py | -rw-r--r-- | 8.8 KB |
| lax_numpy_indexing_test.py | -rw-r--r-- | 55.1 KB |
| lax_numpy_test.py | -rw-r--r-- | 280.6 KB |
| lax_numpy_vectorize_test.py | -rw-r--r-- | 7.5 KB |
| lax_scipy_sparse_test.py | -rw-r--r-- | 17.2 KB |
| lax_scipy_test.py | -rw-r--r-- | 24.6 KB |
| lax_test.py | -rw-r--r-- | 132.0 KB |
| lax_vmap_test.py | -rw-r--r-- | 36.8 KB |
| linalg_test.py | -rw-r--r-- | 62.9 KB |
| loops_test.py | -rw-r--r-- | 14.7 KB |
| masking_test.py | -rw-r--r-- | 30.7 KB |
| mesh_utils_test.py | -rw-r--r-- | 11.2 KB |
| metadata_test.py | -rw-r--r-- | 4.1 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.7 KB |
| nn_test.py | -rw-r--r-- | 10.1 KB |
| ode_test.py | -rw-r--r-- | 7.9 KB |
| optimizers_test.py | -rw-r--r-- | 10.1 KB |
| pickle_test.py | -rw-r--r-- | 1.8 KB |
| pjit_test.py | -rw-r--r-- | 57.4 KB |
| pmap_test.py | -rw-r--r-- | 98.3 KB |
| polynomial_test.py | -rw-r--r-- | 5.7 KB |
| profiler_test.py | -rw-r--r-- | 6.2 KB |
| qdwh_test.py | -rw-r--r-- | 7.8 KB |
| random_test.py | -rw-r--r-- | 56.7 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.0 KB |
| scipy_optimize_test.py | -rw-r--r-- | 6.8 KB |
| scipy_signal_test.py | -rw-r--r-- | 16.0 KB |
| scipy_stats_test.py | -rw-r--r-- | 22.6 KB |
| sharded_jit_test.py | -rw-r--r-- | 17.0 KB |
| sparse_test.py | -rw-r--r-- | 89.2 KB |
| sparsify_test.py | -rw-r--r-- | 12.4 KB |
| stax_test.py | -rw-r--r-- | 10.8 KB |
| svd_test.py | -rw-r--r-- | 4.8 KB |
| transfer_guard_test.py | -rw-r--r-- | 8.7 KB |
| tree_util_test.py | -rw-r--r-- | 19.0 KB |
| util_test.py | -rw-r--r-- | 2.5 KB |
| version_test.py | -rw-r--r-- | 1.8 KB |
| x64_context_test.py | -rw-r--r-- | 5.2 KB |
| xla_bridge_test.py | -rw-r--r-- | 8.0 KB |
| xla_interpreter_test.py | -rw-r--r-- | 1.1 KB |
| xmap_test.py | -rw-r--r-- | 66.6 KB |
Computing file changes ...
