TensorFlow whitepaper说它有一个用C ++编写的核心。这是否意味着Python中的指定计算图完全转换为执行的C ++等效代码?如果是,是否可以提取生成的中间代码?我的用例是观察cuDNN库对指定计算图的调用。
答案 0 :(得分:1)
如果执行print(tf.get_default_graph().as_graph_def())
,您可以看到中间格式。要观察CuDNN调用,您可以向tensorflow/stream_executor/cuda/cuda_dnn.cc