TF计算图优化完成后,在Session开始执行后就轮到TF 计算图的执行了。Tensorflow会根据计算图的节点信息,首先找到一个没有输入的节点作为根节点,创建一个task交给线程池取执行。每个节点完成后会根据Edge通知下游节点计算,直到所有节点完成计算,然后输出结果。
计算图执行
还是以“TF 生成计算图”中的程序为例:
进入到如下的堆栈,从DirectSession::CreateExecutors()开始创建Executors。
tensorflow::DirectSession::CreateExecutors() at direct_session.cc:1,301 0x7ffff591344b tensorflow::DirectSession::GetOrCreateExecutors() at direct_session.cc:1,435 0x7ffff5914bec tensorflow::DirectSession::PRunSetup() at direct_session.cc:849 0x7ffff59162b7 TF_SessionPRunSetup() at c_api.cc:2,668 0x7ffff2090e54 main() at TensorflowTest2.cpp:58 0x40522e |
继续阅读“TF 计算图的执行”本作品采用知识共享署名 4.0 国际许可协议进行许可。