Searched refs:set_allowed_devices (Results 1 – 8 of 8) sorted by relevance
63 LocalClientOptions& LocalClientOptions::set_allowed_devices( in set_allowed_devices() function in xla::LocalClientOptions86 default_options.set_allowed_devices(device_set); in GetOrCreateLocalClient()111 service_options.set_allowed_devices(options.allowed_devices()); in GetOrCreateLocalClient()
68 LocalClientOptions& set_allowed_devices(
60 BackendOptions& set_allowed_devices(
59 BackendOptions& BackendOptions::set_allowed_devices( in set_allowed_devices() function in xla::BackendOptions
58 .set_allowed_devices(options.allowed_devices()); in NewService()
67 ServiceOptions& set_allowed_devices(
124 ServiceOptions& ServiceOptions::set_allowed_devices( in set_allowed_devices() function in xla::ServiceOptions150 backend_options.set_allowed_devices(options.allowed_devices()); in NewService()
247 .set_allowed_devices(allowed_devices_) in GetOrCreateClient()