Home
last modified time | relevance | path

Searched defs:device_assignment_ (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/
Dexecutable_run_options.h161 const DeviceAssignment* device_assignment_ = nullptr; variable
/external/tensorflow/tensorflow/compiler/xla/client/
Dexecutable_build_options.h146 absl::optional<DeviceAssignment> device_assignment_; variable
/external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/client/
Dtpu_client.h375 const DeviceAssignment device_assignment_; variable
/external/tensorflow/tensorflow/compiler/xla/pjrt/
Dtfrt_cpu_pjrt_client.h615 std::shared_ptr<DeviceAssignment> device_assignment_; variable
Dpjrt_stream_executor_client.h801 std::shared_ptr<DeviceAssignment> device_assignment_; variable