Home
last modified time | relevance | path

Searched defs:device_assignment_ (Results 1 – 4 of 4) 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.h152 absl::optional<DeviceAssignment> device_assignment_; variable
/external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/client/
Dtpu_client.h367 const DeviceAssignment device_assignment_; variable
/external/tensorflow/tensorflow/compiler/xla/pjrt/
Dpjrt_stream_executor_client.h715 std::shared_ptr<DeviceAssignment> device_assignment_; variable