Oleg Shyshkov fb80d2abcb [XLA][NFC] Make interface of module loaders consistent. ...

LoadModuleFromData has (data, format, config, ...) signature while FromFile has (path, config, format, ...). Change the latter so `format` becomes the second argument in both cases.

Since I'm touching this file:
* Use `std::string_view` and `absl::Status`
* Change `ovr_config` parameter to `const &`

PiperOrigin-RevId: 601304308

2024-01-24 19:16:43 -08:00

..

jax_cpp

[XLA][NFC] Make interface of module loaders consistent.

2024-01-24 19:16:43 -08:00

__init__.py

Change JAX's copyright attribution from "Google LLC" to "The JAX Authors.".

2022-09-22 12:27:19 -07:00

advi.py

Change JAX's copyright attribution from "Google LLC" to "The JAX Authors.".

2022-09-22 12:27:19 -07:00

datasets.py

Change JAX's copyright attribution from "Google LLC" to "The JAX Authors.".

2022-09-22 12:27:19 -07:00

differentially_private_sgd.py

Update flags to use the ABSL typed flag API.

2023-07-27 12:15:58 -07:00

examples_test.py

Change to simpler import for jax.config

2023-04-21 11:51:22 -07:00

gaussian_process_regression.py

Fix for log-normal prior in example.

2023-08-09 13:38:37 +01:00

kernel_lsq.py

Change JAX's copyright attribution from "Google LLC" to "The JAX Authors.".

2022-09-22 12:27:19 -07:00

mnist_classifier_fromscratch.py

Change JAX's copyright attribution from "Google LLC" to "The JAX Authors.".

2022-09-22 12:27:19 -07:00

mnist_classifier.py

Change JAX's copyright attribution from "Google LLC" to "The JAX Authors.".

2022-09-22 12:27:19 -07:00

mnist_vae.py

Change JAX's copyright attribution from "Google LLC" to "The JAX Authors.".

2022-09-22 12:27:19 -07:00

onnx2xla.py

Change JAX's copyright attribution from "Google LLC" to "The JAX Authors.".

2022-09-22 12:27:19 -07:00

spmd_mnist_classifier_fromscratch.py

Change JAX's copyright attribution from "Google LLC" to "The JAX Authors.".

2022-09-22 12:27:19 -07:00