Home
last modified time | relevance | path

Searched refs:operation_id_ (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/
Dpod_tpu_driver.cc49 : driver_(driver), operation_id_(operation_id) {} in PodEvent()
50 int64_t operation_id() const { return operation_id_; } in operation_id()
61 const int64_t operation_id_; member in tpu_driver::__anon2989b0910111::PodEvent
167 operation_id_(operation_id), in PodBufferHandle()
170 event_(std::make_shared<PodEvent>(driver_, operation_id_)), in PodBufferHandle()
177 int64_t operation_id() const { return operation_id_; } in operation_id()
182 const int64_t operation_id_; member in tpu_driver::__anon2989b0910111::PodBufferHandle
193 operation_id_(operation_id), in PodCompiledProgramHandle()
194 event_(std::make_shared<PodEvent>(driver_, operation_id_)) {} in PodCompiledProgramHandle()
200 int64_t operation_id() const { return operation_id_; } in operation_id()
[all …]
Dgrpc_tpu_driver.cc420 EventId NewOperationId() { return EventId{client_id_, ++operation_id_}; } in NewOperationId()
439 std::atomic<uint64_t> operation_id_{0}; member in tpu_driver::__anond0091eda0111::GrpcTpuDriver