如何在TensorFlow中试验放置算法

Question

我知道TF没有实现他们在白皮书中提出的节点放置算法（尚未）。 TF目前只提供'simple_placer'。

如果我想尝试不同的算法和成本模型，有没有一种简单的方法可以实现/测试它而无需重新编译整个TF并用Pip重新安装它？

Answer 1

您可以使用Python设备setter函数来决定在Python图形构建期间放置节点的位置。

示例：

1. https://stackoverflow.com/a/39991649/419116
2. https://github.com/tensorflow/tensorflow/blob/e7508b6efd51f7cfed32ff277e805f8ea60a7a48/tensorflow/contrib/training/python/training/device_setter_test.py