jax
4b033433 - Add pmap_shard_device_array_benchmark. (#2864)

Commit

6 years ago

Add pmap_shard_device_array_benchmark. (#2864) Also renames pmap_shard_args_benchmark to pmap_shard_sharded_device_array_benchmark.

References

Author

skye

Parents