Searched defs:set_device_allocator (Results 1 – 2 of 2) sorted by relevance
25 ExecutableBuildOptions& ExecutableBuildOptions::set_device_allocator( in set_device_allocator() function in xla::ExecutableBuildOptions
89 void set_device_allocator(se::DeviceMemoryAllocator* device_allocator) { in set_device_allocator() function