在TensorFlow

Question

添加新設備在TensorFlow，一個會話地方的圖表OPS到設備，如CPU S或GPU s和提供方法來執行它們。是否可以向TensorFlow添加額外的設備類？除了CPU和GPU之外，我們稱之爲XPU？

Answer 1

雖然我們沒有很好的文檔，但可以向TensorFlow添加新的設備類型。約，將需要新增一個新的實現的tensorflow::Device，加入tensorflow::DeviceFactory一個相應的實現，並調用以下注冊宏：

REGISTER_LOCAL_DEVICE_FACTORY("XPU", MyXPUDeviceFactory);