Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_rematerialization.cc866 Item* best_item = nullptr; in PickRematerializationCandidate() local
923 if (best_item == nullptr || cost < best_cost) { in PickRematerializationCandidate()
925 best_item = item; in PickRematerializationCandidate()
929 return best_item; in PickRematerializationCandidate()
1031 Item* best_item = PickRematerializationCandidate( in RematerializeComputation() local
1034 if (best_item == nullptr) { in RematerializeComputation()
1043 HloInstruction* best = best_item->instruction; in RematerializeComputation()
1046 memory_tracker.MemoryReducedIfRematerialized(best_item)) in RematerializeComputation()
1076 memory_tracker.AddRematerializedInstruction(best_item, remat_item)); in RematerializeComputation()