HomeSort by relevance Sort by last modified time
    Searched refs:TemporaryMemoryManager (Results 1 - 4 of 4) sorted by null

  /external/tensorflow/tensorflow/stream_executor/
temporary_memory_manager.h 37 // Record used inside the TemporaryMemoryManager as metadata for a given device
56 class TemporaryMemoryManager {
58 explicit TemporaryMemoryManager(Stream* stream) : stream_(stream) {}
128 SE_DISALLOW_COPY_AND_ASSIGN(TemporaryMemoryManager);
136 TemporaryMemoryManager::AllocateArray(uint64 element_count) {
temporary_memory_manager.cc 28 void TemporaryMemoryManager::ForceDeallocateAll() {
37 void TemporaryMemoryManager::MarkFinalized(
51 void TemporaryMemoryManager::DeallocateFinalizedTemporaries() {
67 bool TemporaryMemoryManager::IsFinalized(const DeviceMemoryBase& device_memory,
83 bool TemporaryMemoryManager::HasAllocated(const DeviceMemoryBase& device_memory,
94 TemporaryMemoryManager::AllocateArrayBase(uint64 element_count,
temporary_device_memory.h 50 class TemporaryMemoryManager;
88 friend class internal::TemporaryMemoryManager;
stream.h     [all...]

Completed in 230 milliseconds