TensorFlow数据流图
2019-08-01 13:58:44 0 举报
Tensorflow生成数据流图的源码剖析流程
作者其他创作
大纲/内容
inline_ready.push_back(tagged_node)
Session.Run()
异步与同步调用不同api
处理上下文,运行参数等
tensorflow:::DirectSession::Run()
累加计数器
其他的node扔进线程池
inline_ready->push_back(*curr_expensive_node)
ExecutorState::RunAsync()
运行Process的内容
初始化ready队列,将入度为0的点push进去
将ready转移自己要做的任务到inline_ready,保证线程只做一个expensive node
RunState标记
NodeDone()
让Exec配合线程池工作
循环做inline_ready中的node
处理输出
inline_ready=nullptr
设置性能追踪工具等
处理输入、准备线程池、检查Executor
收藏
0 条评论
下一页