Searched defs:memory_allocator (Results 1 – 11 of 11) sorted by relevance
65 DeviceMemoryAllocator* memory_allocator() const { return memory_allocator_; } in memory_allocator() function87 DeviceMemoryAllocator* memory_allocator, in BufferAllocations()
32 ScratchAllocator(int device_ordinal, DeviceMemoryAllocator* memory_allocator) in ScratchAllocator()
42 DeviceMemoryAllocator* memory_allocator) { in Build()
32 int device_ordinal, DeviceMemoryAllocator* memory_allocator) in FftScratchAllocator()
240 DeviceMemoryAllocator* memory_allocator = run_options->allocator(); in ExecuteOnStream() local
91 DeviceMemoryAllocator* memory_allocator() const { in memory_allocator() function
157 DeviceMemoryAllocator* memory_allocator() const { return allocator_; } in memory_allocator() function
78 DeviceMemoryAllocator* memory_allocator, int device_ordinal, in CreateBufferTable()300 DeviceMemoryAllocator* memory_allocator = run_options->allocator(); in ExecuteAsyncOnStreamImpl() local
217 PersistentMemoryAllocator* memory_allocator() { in memory_allocator() function
914 PersistentMemoryAllocator* memory_allocator = in ReleaseForTesting() local
792 MemoryAllocator* memory_allocator() { return memory_allocator_; } in memory_allocator() function