Searched defs:memory_allocator_ (Results 1 – 9 of 9) sorted by relevance
52 se::DeviceMemoryAllocator* memory_allocator_; variable
109 se::DeviceMemoryAllocator* memory_allocator_; variable
73 se::DeviceMemoryAllocator* memory_allocator_; member in xla::gpu::__anona42065910111::ScratchAllocator
73 SimpleMemoryAllocator* memory_allocator_; variable
76 SimpleMemoryAllocator* memory_allocator_; member in tflite::__anon9074a4740111::MicroBuiltinDataAllocator
182 std::unique_ptr<se::StreamExecutorMemoryAllocator> memory_allocator_; variable
113 DeviceMemoryAllocator* memory_allocator_; variable
318 std::unique_ptr<PersistentMemoryAllocator> memory_allocator_; variable
2199 std::unique_ptr<MemoryAllocator> memory_allocator_; variable