Searched defs:memory_allocator (Results 1 – 16 of 16) sorted by relevance
65 se::DeviceMemoryAllocator* memory_allocator() const { in memory_allocator() function89 se::DeviceMemoryAllocator* memory_allocator, in BufferAllocations()
42 se::DeviceMemoryAllocator* memory_allocator) { in Build()
32 int device_ordinal, se::DeviceMemoryAllocator* memory_allocator) in FftScratchAllocator()
306 se::DeviceMemoryAllocator* const memory_allocator = run_options->allocator(); in ExecuteAsyncOnStream() local
54 se::DeviceMemoryAllocator* memory_allocator) in ScratchAllocator()
91 se::DeviceMemoryAllocator* memory_allocator() const { in memory_allocator() function
159 se::DeviceMemoryAllocator* memory_allocator() const { return allocator_; } in memory_allocator() function
82 se::DeviceMemoryAllocator* memory_allocator, int device_ordinal, in CreateBufferTable()313 se::DeviceMemoryAllocator* memory_allocator = run_options->allocator(); in ExecuteAsyncOnStream() local
285 void SharedReadOnlySpace::TearDown(MemoryAllocator* memory_allocator) { in TearDown()294 void ReadOnlySpace::TearDown(MemoryAllocator* memory_allocator) { in TearDown()321 void ReadOnlySpace::SetPermissionsForPages(MemoryAllocator* memory_allocator, in SetPermissionsForPages()377 auto* memory_allocator = heap()->memory_allocator(); in Seal() local
21 MemoryAllocator* memory_allocator = heap_->memory_allocator(); in DiscardUnusedMemory() local
850 MemoryAllocator* memory_allocator() { return memory_allocator_.get(); } in memory_allocator() function851 const MemoryAllocator* memory_allocator() const { in memory_allocator() function
217 PersistentMemoryAllocator* memory_allocator() { in memory_allocator() function
914 PersistentMemoryAllocator* memory_allocator = in ReleaseForTesting() local
73 auto memory_allocator = in RunAndCheck() local
48 DeviceMemoryAllocator* memory_allocator, in RedzoneAllocator()
61 explicit MicroBuiltinDataAllocator(SimpleMemoryAllocator* memory_allocator) in MicroBuiltinDataAllocator()