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.cc1536 Item* best_item = (*best_items)[i]; in RematerializeInstructions() local
1537 HloInstruction* best = best_item->instruction; in RematerializeInstructions()
1544 if (!memory_tracker->HasUnplacedUsers(best_item)) { in RematerializeInstructions()
1564 for (auto& user : memory_tracker->GetItemUses(best_item)) { in RematerializeInstructions()
1597 best_item, remat_item, absl::MakeSpan(indirect_users))); in RematerializeInstructions()
1683 Item* best_item, const Shape& compact_shape, in CompressInstruction() argument
1685 HloInstruction* best = best_item->instruction; in CompressInstruction()
1688 best_item, compact_shape)) in CompressInstruction()
1717 best_item, compressed_item, uncompressed_item)); in CompressInstruction()
1731 instruction_list->InsertAfterInstructions(compressed_item, {best_item}); in CompressInstruction()