Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_rematerialization.cc652 int64 memory_reduced = 0; in MemoryReducedIfRematerialized() local
663 memory_reduced += AllocatedSize(buffer_id); in MemoryReducedIfRematerialized()
674 memory_reduced -= AllocatedSize(buffer_id); in MemoryReducedIfRematerialized()
678 return memory_reduced; in MemoryReducedIfRematerialized()
841 int64 memory_reduced, int64 memory_limit_bytes) { in RematerializationCost() argument
852 CHECK_GT(memory_reduced, 0); in RematerializationCost()
854 return memory_limit_bytes / memory_reduced; in RematerializationCost()
908 const int64 memory_reduced = in PickRematerializationCandidate() local
911 if (memory_reduced <= 0) { in PickRematerializationCandidate()
913 << " memory reduced = " << memory_reduced << " <= 0"; in PickRematerializationCandidate()
[all …]