xla

71999f59 - Cache OpSharding by mesh and partition_spec (#5593)

Cache OpSharding by mesh and partition_spec (#5593) * Cache OpSharding in mark_sharding * Cache global_runtime_device_count * Add metric and unit test * Use functools.lru_cache in get_op_sharding * yapf