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.cc28 void TemporaryMemoryManager::ForceDeallocateAll() { in ForceDeallocateAll()
37 void TemporaryMemoryManager::MarkFinalized( in MarkFinalized()
51 void TemporaryMemoryManager::DeallocateFinalizedTemporaries() { in DeallocateFinalizedTemporaries()
67 bool TemporaryMemoryManager::IsFinalized(const DeviceMemoryBase& device_memory, in IsFinalized()
83 bool TemporaryMemoryManager::HasAllocated(const DeviceMemoryBase& device_memory, in HasAllocated()
94 TemporaryMemoryManager::AllocateArrayBase(uint64 element_count, in AllocateArrayBase()
Dtemporary_memory_manager.h56 class TemporaryMemoryManager {
58 explicit TemporaryMemoryManager(Stream* stream) : stream_(stream) {} in TemporaryMemoryManager() function
128 SE_DISALLOW_COPY_AND_ASSIGN(TemporaryMemoryManager);
136 TemporaryMemoryManager::AllocateArray(uint64 element_count) { in AllocateArray()
Dtemporary_device_memory.h50 class TemporaryMemoryManager; variable
88 friend class internal::TemporaryMemoryManager;
Dstream.h1953 internal::TemporaryMemoryManager *temporary_memory_manager();
2022 internal::TemporaryMemoryManager temporary_memory_manager_;
2044 inline internal::TemporaryMemoryManager *Stream::temporary_memory_manager() { in temporary_memory_manager()