Trong TensorFlow, một phiên nơi các ops graph vào Devices, chẳng hạn như CPU s hoặc GPU s, và cung cấp phương pháp để thực hiện chúng. Có thể thêm một lớp thiết bị bổ sung vào TensorFlow không? Hãy gọi nó là XPU ngoài CPU và GPU?Thêm một thiết bị mới trong TensorFlow
Có thể thêm các loại thiết bị mới vào TensorFlow, mặc dù chúng tôi không có tài liệu tuyệt vời. Khoảng, nó sẽ đòi hỏi thêm một thực hiện mới của tensorflow::Device, thêm một thi tương ứng của tensorflow::DeviceFactory, và cách gọi vĩ mô đăng ký sau:
REGISTER_LOCAL_DEVICE_FACTORY("XPU", MyXPUDeviceFactory);
Việc khởi tạo tài liệu như vậy có trong yêu cầu kéo này/https://github.com/tensorflow/tensorflow/pull/4305 –
Hi, bạn đã tìm thấy một cách để thêm XPU của bạn để TensorFlow? –