Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dmemory_space_assignment.cc639 void CostAnalysisPrefetchIntervalPicker::SetRetryNumber(int retry_number) { in SetRetryNumber() argument
641 max_overlap_multiplier_ = 1 << retry_number; in SetRetryNumber()
1201 for (int retry_number = 0; retry_number < options_.max_retries; in Finish() local
1202 retry_number++) { in Finish()
1204 bool final_retry = (retry_number == options_.max_retries - 1); in Finish()
1205 options_.prefetch_interval_picker->SetRetryNumber(retry_number); in Finish()
1213 VLOG(2) << "Couldn't allocate. Retry number " << retry_number; in Finish()
1231 --retry_number; in Finish()
2399 for (int retry_number = 0; in FindPrefetchEndTime() local
2400 retry_number < options_.prefetch_copy_done_reorder_max_retries; in FindPrefetchEndTime()
[all …]
Dmemory_space_assignment.h227 virtual void SetRetryNumber(int retry_number) {} in SetRetryNumber() argument
337 void SetRetryNumber(int retry_number) override;