Searched defs:memory_allocator (Results 1 – 12 of 12) sorted by relevance
40 se::DeviceMemoryAllocator* memory_allocator) in BufferAllocations()50 se::DeviceMemoryAllocator* memory_allocator() const { in memory_allocator() function
320 se::DeviceMemoryAllocator* const memory_allocator, int device_ordinal, in BufferForAllocation()391 se::DeviceMemoryAllocator* const memory_allocator, in GenerateBufferAllocations()603 se::DeviceMemoryAllocator* const memory_allocator = run_options->allocator(); in ExecuteAsyncOnStreamImpl() local
31 int device_ordinal, se::DeviceMemoryAllocator* memory_allocator) in FftScratchAllocator()
59 se::DeviceMemoryAllocator* memory_allocator) in ScratchAllocator()
95 se::DeviceMemoryAllocator* memory_allocator, int device_ordinal) { in MemoryForAllocation()129 se::DeviceMemoryAllocator* memory_allocator, int device_ordinal, in CreateBufferTable()355 se::DeviceMemoryAllocator* memory_allocator = run_options->allocator(); in ExecuteAsyncOnStream() local
91 se::DeviceMemoryAllocator* memory_allocator() const { in memory_allocator() function
176 se::DeviceMemoryAllocator* memory_allocator() const { return allocator_; } in memory_allocator() function
217 PersistentMemoryAllocator* memory_allocator() { in memory_allocator() function
914 PersistentMemoryAllocator* memory_allocator = in ReleaseForTesting() local
49 DeviceMemoryAllocator* memory_allocator, in RedzoneAllocator()
82 se::StreamExecutorMemoryAllocator memory_allocator( in RunAndCheck() local
385 se::DeviceMemoryAllocator* const memory_allocator; member