TensorFlow Python前端与其C ++核心之间的交互

时间:2016-12-07 02:55:25

标签: tensorflow

TensorFlow whitepaper说它有一个用C ++编写的核心。这是否意味着Python中的指定计算图完全转换为执行的C ++等效代码?如果是,是否可以提取生成的中间代码?我的用例是观察cuDNN库对指定计算图的调用。

1 个答案:

答案 0 :(得分:1)

如果执行print(tf.get_default_graph().as_graph_def()),您可以看到中间格式。要观察CuDNN调用,您可以向tensorflow/stream_executor/cuda/cuda_dnn.cc

添加一些打印语句