| .. |
|
__pycache__
|
feature: "ANN commit 2"
|
2023-06-19 00:49:18 +02:00 |
|
clusters
|
feature: "ANN commit 2"
|
2023-06-19 00:49:18 +02:00 |
|
debugger
|
feature: "ANN commit 2"
|
2023-06-19 00:49:18 +02:00 |
|
image
|
feature: "ANN commit 2"
|
2023-06-19 00:49:18 +02:00 |
|
internal_test_util
|
feature: "ANN commit 2"
|
2023-06-19 00:49:18 +02:00 |
|
interpreters
|
feature: "ANN commit 2"
|
2023-06-19 00:49:18 +02:00 |
|
lax
|
feature: "ANN commit 2"
|
2023-06-19 00:49:18 +02:00 |
|
lib
|
feature: "ANN commit 2"
|
2023-06-19 00:49:18 +02:00 |
|
nn
|
feature: "ANN commit 2"
|
2023-06-19 00:49:18 +02:00 |
|
numpy
|
feature: "ANN commit 2"
|
2023-06-19 00:49:18 +02:00 |
|
ops
|
feature: "ANN commit 2"
|
2023-06-19 00:49:18 +02:00 |
|
scipy
|
feature: "ANN commit 2"
|
2023-06-19 00:49:18 +02:00 |
|
state
|
feature: "ANN commit 2"
|
2023-06-19 00:49:18 +02:00 |
|
third_party
|
feature: "ANN commit 2"
|
2023-06-19 00:49:18 +02:00 |
|
__init__.py
|
feature: "ANN commit 2"
|
2023-06-19 00:49:18 +02:00 |
|
abstract_arrays.py
|
feature: "ANN commit 2"
|
2023-06-19 00:49:18 +02:00 |
|
ad_checkpoint.py
|
feature: "ANN commit 2"
|
2023-06-19 00:49:18 +02:00 |
|
ad_util.py
|
feature: "ANN commit 2"
|
2023-06-19 00:49:18 +02:00 |
|
api_util.py
|
feature: "ANN commit 2"
|
2023-06-19 00:49:18 +02:00 |
|
api.py
|
feature: "ANN commit 2"
|
2023-06-19 00:49:18 +02:00 |
|
array.py
|
feature: "ANN commit 2"
|
2023-06-19 00:49:18 +02:00 |
|
basearray.py
|
feature: "ANN commit 2"
|
2023-06-19 00:49:18 +02:00 |
|
basearray.pyi
|
feature: "ANN commit 2"
|
2023-06-19 00:49:18 +02:00 |
|
callback.py
|
feature: "ANN commit 2"
|
2023-06-19 00:49:18 +02:00 |
|
checkify.py
|
feature: "ANN commit 2"
|
2023-06-19 00:49:18 +02:00 |
|
cloud_tpu_init.py
|
feature: "ANN commit 2"
|
2023-06-19 00:49:18 +02:00 |
|
compilation_cache_interface.py
|
feature: "ANN commit 2"
|
2023-06-19 00:49:18 +02:00 |
|
compilation_cache.py
|
feature: "ANN commit 2"
|
2023-06-19 00:49:18 +02:00 |
|
config.py
|
feature: "ANN commit 2"
|
2023-06-19 00:49:18 +02:00 |
|
core.py
|
feature: "ANN commit 2"
|
2023-06-19 00:49:18 +02:00 |
|
custom_api_util.py
|
feature: "ANN commit 2"
|
2023-06-19 00:49:18 +02:00 |
|
custom_batching.py
|
feature: "ANN commit 2"
|
2023-06-19 00:49:18 +02:00 |
|
custom_derivatives.py
|
feature: "ANN commit 2"
|
2023-06-19 00:49:18 +02:00 |
|
custom_transpose.py
|
feature: "ANN commit 2"
|
2023-06-19 00:49:18 +02:00 |
|
debugging.py
|
feature: "ANN commit 2"
|
2023-06-19 00:49:18 +02:00 |
|
deprecations.py
|
feature: "ANN commit 2"
|
2023-06-19 00:49:18 +02:00 |
|
dispatch.py
|
feature: "ANN commit 2"
|
2023-06-19 00:49:18 +02:00 |
|
distributed.py
|
feature: "ANN commit 2"
|
2023-06-19 00:49:18 +02:00 |
|
dlpack.py
|
feature: "ANN commit 2"
|
2023-06-19 00:49:18 +02:00 |
|
dtypes.py
|
feature: "ANN commit 2"
|
2023-06-19 00:49:18 +02:00 |
|
effects.py
|
feature: "ANN commit 2"
|
2023-06-19 00:49:18 +02:00 |
|
environment_info.py
|
feature: "ANN commit 2"
|
2023-06-19 00:49:18 +02:00 |
|
errors.py
|
feature: "ANN commit 2"
|
2023-06-19 00:49:18 +02:00 |
|
flatten_util.py
|
feature: "ANN commit 2"
|
2023-06-19 00:49:18 +02:00 |
|
gfile_cache.py
|
feature: "ANN commit 2"
|
2023-06-19 00:49:18 +02:00 |
|
iree.py
|
feature: "ANN commit 2"
|
2023-06-19 00:49:18 +02:00 |
|
lax_reference.py
|
feature: "ANN commit 2"
|
2023-06-19 00:49:18 +02:00 |
|
lazy_loader.py
|
feature: "ANN commit 2"
|
2023-06-19 00:49:18 +02:00 |
|
linear_util.py
|
feature: "ANN commit 2"
|
2023-06-19 00:49:18 +02:00 |
|
maps.py
|
feature: "ANN commit 2"
|
2023-06-19 00:49:18 +02:00 |
|
mesh.py
|
feature: "ANN commit 2"
|
2023-06-19 00:49:18 +02:00 |
|
monitoring.py
|
feature: "ANN commit 2"
|
2023-06-19 00:49:18 +02:00 |
|
op_shardings.py
|
feature: "ANN commit 2"
|
2023-06-19 00:49:18 +02:00 |
|
partition_spec.py
|
feature: "ANN commit 2"
|
2023-06-19 00:49:18 +02:00 |
|
path.py
|
feature: "ANN commit 2"
|
2023-06-19 00:49:18 +02:00 |
|
pjit.py
|
feature: "ANN commit 2"
|
2023-06-19 00:49:18 +02:00 |
|
pretty_printer.py
|
feature: "ANN commit 2"
|
2023-06-19 00:49:18 +02:00 |
|
prng.py
|
feature: "ANN commit 2"
|
2023-06-19 00:49:18 +02:00 |
|
profiler.py
|
feature: "ANN commit 2"
|
2023-06-19 00:49:18 +02:00 |
|
public_test_util.py
|
feature: "ANN commit 2"
|
2023-06-19 00:49:18 +02:00 |
|
random.py
|
feature: "ANN commit 2"
|
2023-06-19 00:49:18 +02:00 |
|
sharding_impls.py
|
feature: "ANN commit 2"
|
2023-06-19 00:49:18 +02:00 |
|
sharding_specs.py
|
feature: "ANN commit 2"
|
2023-06-19 00:49:18 +02:00 |
|
sharding.py
|
feature: "ANN commit 2"
|
2023-06-19 00:49:18 +02:00 |
|
source_info_util.py
|
feature: "ANN commit 2"
|
2023-06-19 00:49:18 +02:00 |
|
stages.py
|
feature: "ANN commit 2"
|
2023-06-19 00:49:18 +02:00 |
|
test_util.py
|
feature: "ANN commit 2"
|
2023-06-19 00:49:18 +02:00 |
|
traceback_util.py
|
feature: "ANN commit 2"
|
2023-06-19 00:49:18 +02:00 |
|
tree_util.py
|
feature: "ANN commit 2"
|
2023-06-19 00:49:18 +02:00 |
|
typing.py
|
feature: "ANN commit 2"
|
2023-06-19 00:49:18 +02:00 |
|
util.py
|
feature: "ANN commit 2"
|
2023-06-19 00:49:18 +02:00 |
|
xla_bridge.py
|
feature: "ANN commit 2"
|
2023-06-19 00:49:18 +02:00 |
