Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_rematerialization.cc459 return item->placed && item != in_progress_item_; in IsFinished()
465 if (in_progress_item_ == nullptr) { in IsInUse()
468 const BufferIdList& in_progress_uses = in_progress_item_->buffers_used; in IsInUse()
502 Item* in_progress_item_ = nullptr; member in xla::__anon6c3ee80d0111::MemoryUsageTracker
573 TF_RET_CHECK(in_progress_item_ == nullptr); in BeginInstruction()
574 in_progress_item_ = item; in BeginInstruction()
598 TF_RET_CHECK(in_progress_item_ != nullptr); in EndInstruction()
599 VLOG(3) << "EndInstruction " << in_progress_item_->instruction->name(); in EndInstruction()
601 for (BufferId buffer_id : in_progress_item_->buffers_used) { in EndInstruction()
616 for (BufferId buffer_id : in_progress_item_->buffers_defined) { in EndInstruction()
[all …]