Searched defs:DeviceMemoryAllocator (Results 1 – 5 of 5) sorted by relevance
32 class DeviceMemoryAllocator; variable159 explicit DeviceMemoryAllocator(const Platform* platform) in DeviceMemoryAllocator() function
44 class DeviceMemoryAllocator; variable
94 se::DeviceMemoryAllocator* /*device_allocator*/) { in RunHloPasses()102 se::DeviceMemoryAllocator* /*device_allocator*/) { in RunBackend()
30 class DeviceMemoryAllocator; variable
511 se::DeviceMemoryAllocator* /*device_allocator*/) { in RunHloPasses()561 se::DeviceMemoryAllocator* /*device_allocator*/) { in RunBackend()