Searched defs:device_allocator (Results 1 – 9 of 9) sorted by relevance
27 DeviceMemoryAllocator* device_allocator) { in RunHloPassesOnModuleGroup()36 DeviceMemoryAllocator* device_allocator) { in RunBackendOnModuleGroup()44 DeviceMemoryAllocator* device_allocator) { in Compile()
78 DeviceMemoryAllocator* device_allocator() const { return device_allocator_; } in device_allocator() function79 void set_device_allocator(DeviceMemoryAllocator* device_allocator) { in set_device_allocator()
350 DeviceMemoryAllocator* device_allocator) { in BuildExecutables()786 se::StreamExecutor* executor, DeviceMemoryAllocator* device_allocator) { in BuildExecutable()
108 DeviceMemoryAllocator* device_allocator) { in RunHloPassesOnModuleGroup()140 DeviceMemoryAllocator* device_allocator) { in RunBackendOnModuleGroup()148 DeviceMemoryAllocator* device_allocator) { in Compile()
30 DeviceMemoryAllocator* ExecutableBuildOptions::device_allocator() const { in device_allocator() function in xla::ExecutableBuildOptions
43 xla::DeviceMemoryAllocator* device_allocator) in XlaPlatformInfo()
65 xla::DeviceMemoryAllocator* device_allocator = nullptr; in PlatformInfoFromContext() local
180 DeviceMemoryAllocator* device_allocator, in OptimizeHloModule()620 DeviceMemoryAllocator* device_allocator) { in RunHloPasses()635 DeviceMemoryAllocator* device_allocator) { in RunBackend()
327 xla::DeviceMemoryAllocator* device_allocator = nullptr; member