如何在 TensorFlow 中试验放置算法

How to experiment with placement algorithm in TensorFlow

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

如果我想尝试不同的算法和成本模型,有没有一种简单的方法可以implement/test无需重新编译整个 TF 并使用 Pip 重新安装它?

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

示例:

  1. https://github.com/tensorflow/tensorflow/blob/e7508b6efd51f7cfed32ff277e805f8ea60a7a48/tensorflow/contrib/training/python/training/device_setter_test.py