Searched defs:device_allocator (Results 1 – 6 of 6) sorted by relevance
77 DeviceMemoryAllocator* device_allocator() const { return device_allocator_; } in device_allocator() function78 void set_device_allocator(DeviceMemoryAllocator* device_allocator) { in set_device_allocator()
27 DeviceMemoryAllocator* device_allocator) { in Compile()
342 DeviceMemoryAllocator* device_allocator) { in BuildExecutables()403 se::StreamExecutor* executor, DeviceMemoryAllocator* device_allocator) { in BuildExecutable()454 DeviceMemoryAllocator* device_allocator) { in BuildAndCacheExecutable()
29 DeviceMemoryAllocator* ExecutableBuildOptions::device_allocator() const { in device_allocator() function in xla::ExecutableBuildOptions
133 DeviceMemoryAllocator* device_allocator) { in OptimizeHloModule()471 DeviceMemoryAllocator* device_allocator) { in RunHloPasses()482 DeviceMemoryAllocator* device_allocator) { in RunBackend()
268 xla::DeviceMemoryAllocator* device_allocator = nullptr; member