テンソルフローGraphDefのSendノードとReceiveノードを確認できますか？

Question

ノードを送信し、ノードをテンソルフローGraphDefに送信するか、python APIを使用することができますか？

私は、次のコード

import tensorflow as tf 

with tf.device("/gpu:0"): 
    x = tf.constant(1.0) 
with tf.device("/gpu:1"): 
    y = tf.constant(2.0) 
with tf.device("/cpu:0"): 
    sum = tf.add(x, y) 

graph_def = tf.get_default_graph().as_graph_def() 
print(graph_def) 

を試してみてください。しかしgraph_defには、送信/ recvのノードがありません。グラフに送信/受信ノードが追加されていますか？xとy〜cpu？

Answer 1

tf.Session.run() ...の呼び出しで初めてグラフを実行しようとすると、sendノードとrecvノードがグラフに追加されます。実際に追加されるsendノードとrecvノードのセットは、あなたがその呼び出しでフィードしてフェッチする特定のテンソルに

次のようにあなたは、Session.run()コールにtf.RunOptions(output_partition_graphs=True)を渡すことによって、送信とrecvのノードを含む各デバイス上で実行正確なグラフを、見ることができます：

options = tf.RunOptions(output_partition_graphs=True) 
metadata = tf.RunMetadata() 

sess.run(..., options=options, metadata=metadata) 

for partition_graph_def in metadata.partition_graphs: 
    print partition_graph_def # Contains all the nodes that ran on a single device.