Searched defs:memory_allocator_ (Results 1 – 6 of 6) sorted by relevance
107 DeviceMemoryAllocator* memory_allocator_; variable
53 DeviceMemoryAllocator* memory_allocator_; variable
52 DeviceMemoryAllocator* memory_allocator_; member in xla::gpu::__anon5d1dd1190111::ScratchAllocator
185 std::unique_ptr<StreamExecutorMemoryAllocator> memory_allocator_; variable
360 std::unique_ptr<PersistentMemoryAllocator> memory_allocator_; variable
2288 MemoryAllocator* memory_allocator_; variable