Searched refs:RematerializationCost (Results 1 – 1 of 1) sorted by relevance
509 int64 RematerializationCost(const std::vector<Item*>& items, in RematerializationCost() function in xla::__anon95ba67740111::MemoryUsageTracker1317 int64 RematerializationCost(const HloInstruction* instruction, in RematerializationCost() function1471 RematerializationCost(block, memory_reduced, memory_limit_bytes); in PickRematerializationCandidates()