Home
last modified time | relevance | path

Searched refs:traceme_context_id (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/batching_util/
Dbatch_scheduler.h83 explicit Batch(uint64 traceme_context_id);
120 uint64 traceme_context_id() const;
205 Batch<TaskType>::Batch(uint64 traceme_context_id) in Batch() argument
206 : traceme_context_id_(traceme_context_id) {} in Batch()
303 uint64 Batch<TaskType>::traceme_context_id() const { in traceme_context_id() function
Dadaptive_shared_batch_scheduler.h309 int64 batch_timeout_micros, uint64 traceme_context_id) in ASBSBatch() argument
313 traceme_context_id_(traceme_context_id) {} in ASBSBatch()
323 uint64 traceme_context_id() const { return traceme_context_id_; } in traceme_context_id() function
541 batch->traceme_context_id()); in CallbackWrapper()
690 this->current_batch_->traceme_context_id()); in Schedule()
747 static std::atomic<uint64> traceme_context_id(0); in NewTraceMeContextIdForBatch()
748 return traceme_context_id.fetch_add(1, std::memory_order_relaxed); in NewTraceMeContextIdForBatch()
Dshared_batch_scheduler.h656 batches_.back()->traceme_context_id()); in ScheduleWithoutSplit()
746 batches_.back()->traceme_context_id()); in ScheduleWithSplit()
821 batch->traceme_context_id()); in ProcessBatch()
/external/tensorflow/tensorflow/core/tpu/kernels/
Dtransfer_ops.cc61 traceme_context_id = schedule_activity.GetContextId()]() { in ComputeAsync()
64 traceme_context_id); in ComputeAsync()