Searched refs:set_device_allocator (Results 1 – 8 of 8) sorted by relevance
72 ExecutableBuildOptions& set_device_allocator(
25 ExecutableBuildOptions& ExecutableBuildOptions::set_device_allocator( in set_device_allocator() function in xla::ExecutableBuildOptions
62 build_options.set_device_allocator(options.device_allocator.get()); in BuildExecutable()
268 build_options.set_device_allocator(options.device_allocator.get()); in GetBuildOptions()
165 build_options.set_device_allocator(device_ref.allocator()); in Compile()
110 void set_device_allocator(se::DeviceMemoryAllocator* device_allocator) { in set_device_allocator() function
362 aot_options.set_device_allocator(options.device_allocator); in BuildAotResults()
2402 build_options.set_device_allocator(allocator()); in GetExecutableExtras()