jax.lib.xla_bridge.get_backend

内容

jax.lib.xla_bridge.get_backend#

jax.lib.xla_bridge.get_backend(platform=None)[source]#
参数:

platform (None | str | xla_client.Client)

返回类型:

xla_client.Client