Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dhlo_memory_scheduler.cc237 int64_t max_subcomputation_bytes = 0; in BytesFreedIfScheduled() local
242 if (subcomputation_bytes > max_subcomputation_bytes) { in BytesFreedIfScheduled()
243 max_subcomputation_bytes = subcomputation_bytes; in BytesFreedIfScheduled()
248 if (max_subcomputation_bytes > 0 && in BytesFreedIfScheduled()
253 bytes_defined = max_subcomputation_bytes; in BytesFreedIfScheduled()
255 bytes_defined = entry.bytes_defined + max_subcomputation_bytes; in BytesFreedIfScheduled()
H A Dheap_simulator.cc471 int64_t max_subcomputation_bytes = 0; in AccountForSubcomputationMemory() local
476 if (subcomputation_bytes > max_subcomputation_bytes) { in AccountForSubcomputationMemory()
477 max_subcomputation_bytes = subcomputation_bytes; in AccountForSubcomputationMemory()
481 if (max_subcomputation_bytes > 0 && in AccountForSubcomputationMemory()
487 max_subcomputation_bytes -= alloc_size_by_instruction; in AccountForSubcomputationMemory()
490 std::max(max_heap_size_, current_heap_size_ + max_subcomputation_bytes); in AccountForSubcomputationMemory()