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.cc126 allowed_devices_ = allowed_devices; in set_allowed_devices()
131 return allowed_devices_; in allowed_devices()
/external/tensorflow/tensorflow/compiler/jit/
Dxla_device.cc214 allowed_devices_(options.allowed_devices) { in XlaDevice()
247 .set_allowed_devices(allowed_devices_) in GetOrCreateClient()
Dxla_device.h273 absl::optional<std::set<int>> allowed_devices_; variable