Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/stream_executor/
Dtemporary_memory_manager.cc27 void TemporaryMemoryManager::ForceDeallocateAll() { in ForceDeallocateAll()
36 void TemporaryMemoryManager::MarkFinalized( in MarkFinalized()
50 void TemporaryMemoryManager::DeallocateFinalizedTemporaries() { in DeallocateFinalizedTemporaries()
66 bool TemporaryMemoryManager::IsFinalized(const DeviceMemoryBase& device_memory, in IsFinalized()
82 bool TemporaryMemoryManager::HasAllocated(const DeviceMemoryBase& device_memory, in HasAllocated()
93 TemporaryMemoryManager::AllocateArrayBase(uint64 element_count, in AllocateArrayBase()
Dtemporary_memory_manager.h56 class TemporaryMemoryManager {
58 explicit TemporaryMemoryManager(Stream* stream) : stream_(stream) {} in TemporaryMemoryManager() function
129 SE_DISALLOW_COPY_AND_ASSIGN(TemporaryMemoryManager);
137 TemporaryMemoryManager::AllocateArray(uint64 element_count) { in AllocateArray()
Dtemporary_device_memory.h50 class TemporaryMemoryManager; variable
88 friend class internal::TemporaryMemoryManager;
Dstream.h2013 internal::TemporaryMemoryManager *temporary_memory_manager();
2085 internal::TemporaryMemoryManager temporary_memory_manager_;
2150 inline internal::TemporaryMemoryManager *Stream::temporary_memory_manager() { in temporary_memory_manager()