| /external/tensorflow/tensorflow/compiler/xla/ |
| D | executable_run_options.h | 51 class RunId {
54 RunId();
55 explicit RunId(int64_t value) : data_(value) {} in RunId() function
57 RunId(const RunId&) = default;
58 RunId& operator=(const RunId&) = default;
59 friend bool operator==(const RunId& a, const RunId& b);
64 friend H AbslHashValue(H h, const RunId& id) { in AbslHashValue()
140 ExecutableRunOptions& set_run_id(RunId id);
141 RunId run_id() const;
169 RunId run_id_;
|
| D | executable_run_options.cc | 22 RunId::RunId() { in RunId() function in xla::RunId
27 bool operator==(const RunId& a, const RunId& b) { return a.data_ == b.data_; } in operator ==()
29 std::string RunId::ToString() const { in ToString()
33 int64_t RunId::ToInt() const { return data_; } in ToInt()
123 ExecutableRunOptions& ExecutableRunOptions::set_run_id(RunId id) { in set_run_id()
128 RunId ExecutableRunOptions::run_id() const { return run_id_; } in run_id()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | collective_ops_utils.h | 165 explicit RendezvousKey(const RunId& run_id, in RendezvousKey()
207 RunId run_id;
|
| D | hlo_runner.cc | 178 nullptr, RunId()); in ExecuteWithDeviceBuffers()
235 RunId run_id; in ExecuteReplicatedImpl()
474 RunId run_id) { in GetServiceRunOptionsForDevice()
|
| D | hlo_runner.h | 156 RunId run_id);
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | nccl_utils.cc | 140 RunId run_id, OpId op_id, NcclCliqueKey clique_key, in AcquireNcclClique()
224 RunId run_id, OpId op_id, std::vector<GlobalDeviceId> participants, in AcquireNcclComm()
|
| D | gpu_executable_run_options.h | 89 RunId run_id;
|
| D | nccl_utils.h | 119 RunId run_id, OpId op_id, std::vector<GlobalDeviceId> participants,
|
| /external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/client/ |
| D | tpu_client.cc | 558 int partition, const RunId& run_id) { in ExecuteHelper()
658 /*replica=*/0, /*partition=*/0, RunId()); in Execute()
726 RunId run_id; in ExecuteOnLocalDevices()
|
| D | tpu_client.h | 431 int partition, const RunId& run_id);
|
| /external/tensorflow/tensorflow/compiler/xla/pjrt/ |
| D | pjrt_stream_executor_client.h | 818 int partition, int executable_idx, const RunId& run_id,
834 const RunId& run_id,
|
| D | tfrt_cpu_pjrt_client.cc | 1305 const RunId& run_id, const ExecuteOptions& options, in ExecuteHelper()
1623 RunId run_id; in Execute()
1747 addressable_device_logical_ids_[i].partition, RunId(), options, in ExecuteSharded()
1785 /*partition=*/0, RunId(), options, in ExecutePortable()
|
| D | pjrt_stream_executor_client.cc | 1901 int executable_idx, const RunId& run_id, const ExecuteOptions& options, in EnqueueExecution()
2093 const RunId& run_id, const ExecuteOptions& options, bool fill_future, in ExecuteHelper()
2203 RunId run_id; in Execute()
2336 addressable_device_logical_ids_[i].partition, RunId(), in ExecuteSharded()
2369 /*partition=*/0, RunId(), in ExecutePortable()
|
| D | tfrt_cpu_pjrt_client.h | 556 int partition, const RunId& run_id, const ExecuteOptions& options,
|
| /external/tensorflow/tensorflow/compiler/xrt/kernels/ |
| D | xrt_execute_op.cc | 319 run_options.set_run_id(xla::RunId(config.run_id())); in RunExecutable()
|
| /external/tensorflow/tensorflow/compiler/jit/kernels/ |
| D | xla_ops.cc | 369 xla::RunId run_id(0); in ComputeAsync()
|