Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/
H A Dexecutable_run_options.h161 const DeviceAssignment* device_assignment_ = nullptr; variable
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/
H A Dexecutable_build_options.h189 std::optional<DeviceAssignment> device_assignment_; variable
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/pjrt/
H A Dtfrt_cpu_pjrt_client.h564 std::shared_ptr<DeviceAssignment> device_assignment_; variable
H A Dpjrt_stream_executor_client.h850 std::shared_ptr<DeviceAssignment> device_assignment_; variable
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/client/
H A Dtpu_client.h435 const DeviceAssignment device_assignment_; variable