在TensorFlow中添加新设备

Question

在TensorFlow,一个会话放置图表OPS到设备,如CPU S或GPU s和提供方法来执行它们.是否可以在TensorFlow中添加其他类设备？让我们称之为XPU以及CPU和GPU？

Answer 1

虽然我们没有很好的文档,但可以向TensorFlow添加新的设备类型.大概需要添加一个新的实现tensorflow::Device,添加相应的实现tensorflow::DeviceFactory,并调用以下注册宏:

REGISTER_LOCAL_DEVICE_FACTORY("XPU", MyXPUDeviceFactory);