Home
last modified time | relevance | path

Searched refs:allowed_devices_ (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/client/
Dclient_library.cc34 allowed_devices_(allowed_devices) {} in LocalClientOptions()
65 allowed_devices_ = allowed_devices; in set_allowed_devices()
71 return allowed_devices_; in allowed_devices()
Dclient_library.h76 absl::optional<std::set<int>> allowed_devices_; variable
/external/tensorflow/tensorflow/compiler/xla/service/
Dbackend.cc61 allowed_devices_ = allowed_devices; in set_allowed_devices()
66 return allowed_devices_; in allowed_devices()
Dbackend.h67 absl::optional<std::set<int>> allowed_devices_; variable
Dservice.h75 absl::optional<std::set<int>> allowed_devices_; variable
Dservice.cc125 allowed_devices_ = allowed_devices; in set_allowed_devices()
130 return allowed_devices_; in allowed_devices()
/external/tensorflow/tensorflow/compiler/jit/
Dxla_device.cc206 allowed_devices_(options.allowed_devices) { in XlaDevice()
236 return xla::ClientLibrary::GetOrCreateLocalClient(platform_, allowed_devices_) in client()
Dxla_device.h250 absl::optional<std::set<int>> allowed_devices_; variable