Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/
H A Dstream.h1587 internal::TemporaryMemoryManager temporary_memory_manager_; variable
1645 return temporary_memory_manager_.AllocateArray<T>(element_count); in AllocateTemporaryArray()
1649 return &temporary_memory_manager_; in temporary_memory_manager()
H A Dstream.cc251 temporary_memory_manager_(this) { in Stream()
264 temporary_memory_manager_.ForceDeallocateAll(); in ~Stream()
2492 temporary_memory_manager_.DeallocateFinalizedTemporaries(); in BlockHostUntilDone()