は、TensorFlowでTensorFlow

Question

に新しいデバイスを追加し、そのようなCPU SまたはGPU Sとしてデバイス、上へセッション場所グラフOPS、およびそれらを実行するための方法を提供します。 TensorFlowに追加のクラスのデバイスを追加することは可能ですか？ CPUとGPUに加えてXPUと呼んでみましょう。

Answer 1

TensorFlowに新しいデバイスタイプを追加することは可能ですが、素晴らしいドキュメントはありません。およそ、それは、tensorflow::Deviceの新しい実装を追加tensorflow::DeviceFactoryの対応する実装を追加し、以下の登録マクロを呼び出す必要になります。

REGISTER_LOCAL_DEVICE_FACTORY("XPU", MyXPUDeviceFactory);