Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_memory_scheduler.cc241 int64 subcomputation_bytes = it->second; in BytesFreedIfScheduled() local
242 if (subcomputation_bytes > max_subcomputation_bytes) { in BytesFreedIfScheduled()
243 max_subcomputation_bytes = subcomputation_bytes; in BytesFreedIfScheduled()
Dheap_simulator.cc472 int64 subcomputation_bytes = it->second; in AccountForSubcomputationMemory() local
473 if (subcomputation_bytes > max_subcomputation_bytes) { in AccountForSubcomputationMemory()
474 max_subcomputation_bytes = subcomputation_bytes; in AccountForSubcomputationMemory()