def _cpu_backend_factory():
  return _xla.get_cpu_client(asynchronous=True)
Exemplo n.º 2
0
def _cpu_backend_factory():
  client = _xla.get_cpu_client(asynchronous=True)
  return LocalBackend(platform='cpu', client=client)