Searched defs:memory_allocator (Results 1 – 12 of 12) sorted by relevance
62 DeviceMemoryAllocator* memory_allocator() const { return memory_allocator_; } in memory_allocator() function88 DeviceMemoryAllocator* memory_allocator) in BufferAllocations()
43 DeviceMemoryAllocator* memory_allocator) { in Build()
33 int device_ordinal, DeviceMemoryAllocator* memory_allocator) in FftScratchAllocator()
254 DeviceMemoryAllocator* memory_allocator = run_options->allocator(); in ExecuteOnStream() local
37 ScratchAllocator(int device_ordinal, DeviceMemoryAllocator* memory_allocator) in ScratchAllocator()
77 DeviceMemoryAllocator* memory_allocator, int device_ordinal, in AllocateBuffers()296 DeviceMemoryAllocator* memory_allocator = run_options->allocator(); in ExecuteOnStream() local328 DeviceMemoryAllocator* memory_allocator = run_options->allocator(); in ExecuteAsyncOnStream() local
327 DeviceMemoryAllocator* memory_allocator, int device_ordinal, in AllocateBuffers()461 DeviceMemoryAllocator* memory_allocator = run_options->allocator(); in ExecuteOnStream() local
137 DeviceMemoryAllocator* memory_allocator() const { return allocator_; } in memory_allocator() function
84 DeviceMemoryAllocator* memory_allocator() const { in memory_allocator() function
217 PersistentMemoryAllocator* memory_allocator() { in memory_allocator() function
831 PersistentMemoryAllocator* memory_allocator = in ReleaseForTesting() local
1039 MemoryAllocator* memory_allocator() { return memory_allocator_; } in memory_allocator() function