![](https://img-blog.csdnimg.cn/20201014180756922.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
tensorflow源代码阅读
sxk20091111
这个作者很懒,什么都没留下…
展开
-
DirectionSession涉及的重要的数据类型
struct ExecutorsAndKeys { ExecutorsAndKeys() : step_count(0) {} std::atomic_int_fast64_t step_count; std::unique_ptr<Graph> graph; NameNodeMap name_to_node; std::vector<PerPartitionExecutorsA...原创 2019-11-12 14:53:26 · 139 阅读 · 0 评论 -
tensorflow StreamExecutor
StreamExecutor框架 Plaftorm Platform是StreamExecutor一个抽象的设备平台层,它是一个抽象类,具体实现的子类有CUDA ,HOST,OpenCL。 通过GetExecutor()就可以获取到相应的StreamExecutor指针。而对于platform的管理是借用MultiPlatformManager来进行管理的。 Mu...原创 2019-10-22 21:09:53 · 592 阅读 · 1 评论