Make make_array_from_process_local_data go via device_put if there is only 1 process.
#61950
copybara-service[bot]ci-build.yaml
on: pull_request
lint_and_typecheck
57s
Matrix: build
Matrix: Documentation - test code snippets
Matrix: Documentation - render documentation
Matrix: jax2tf_test