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.cc582 bool IsInProgressItem(Item* item) const { return item == in_progress_item_; } in IsInProgressItem()
698 return item->placed && item != in_progress_item_; in IsFinished()
704 if (in_progress_item_ == nullptr) { in IsInUse()
707 const BufferIdList& in_progress_uses = in_progress_item_->buffers_used; in IsInUse()
774 Item* in_progress_item_ = nullptr; member in xla::__anon95ba67740111::MemoryUsageTracker
864 TF_RET_CHECK(in_progress_item_ == nullptr); in BeginInstruction()
865 in_progress_item_ = item; in BeginInstruction()
889 TF_RET_CHECK(in_progress_item_ != nullptr); in EndInstruction()
890 VLOG(3) << "EndInstruction " << in_progress_item_->instruction->name(); in EndInstruction()
892 for (BufferId buffer_id : in_progress_item_->buffers_used) { in EndInstruction()
[all …]