TensorFlow技术内幕(三):源码结构

本章讲解TensorFlow的源码结构。

下面的内容我们一TF1.0的代码为例,介绍TF的源码结构,如果是其他版本,则结构可能会略有差异。

根目录结构

首先整体看一下整体的目录结构:

这里写图片描述

图1:TF根目录

然后了解一下每个目录的功能:

目录功能
tensorflow/cC API代码
tensorflow/ccC++ API代码
tensorflow/compilerXLA,JIT等编译优化相关
tensorflow/contribcontributor贡献的代码,这个目录并不是官方支持的, 很有可能在高级 API 完善后被官方迁移到核心的 TensorFlow 目录中或去掉
tensorflow/coretf核心代码
tensorflow/docs_src文档相关文件
tensorflow/examples例子相关代码
tensorflow/g3docTF文档
tensorflow/gogo API相关代码
tensorflow/javajava API相关代码
tensorflow/pythonPython API相关代码
tensorflow/stream_executor并行计算框架代码
tensorflow/tools各种辅助工具工程代码,例如第二章中生成Python安装包的代码就在这里
tensorflow/user_opstf插件代码
third_party/依赖的第三方代码
tools工程编译配置相关
util工程编译相关

表1:TF根目录

其中tensorflow/core是tf的核心模块

核心目录结构

这里写图片描述

图2:TF Core目录

目录功能如下:

目录功能
tensorflow/core/common_runtime公共运行库
tensorflow/core/debug调试相关
tensorflow/core/distributed_runtime分布式执行模块
tensorflow/core/example例子代码
tensorflow/core/framework基础功能模块
tensorflow/core/graph计算图相关
tensorflow/core/grappler模型优化模块
tensorflow/core/kernels操作核心的实现代码,包括CPU和GPU上的实现
tensorflow/core/lib公共基础库
tensorflow/core/ops操作代码
tensorflow/core/platform平台实现相关代码
tensorflow/core/protobuf.proto定义文件
tensorflow/core/publicAPI头文件
tensorflow/core/user_ops
tensorflow/core/util

表2:TF Core目录

小结

本章比较简短,粗略的梳理了一下源码目录结构。在未来章节中,我会围绕具体的主题功能来细致的分析涉及到的源代码。

  • 3
    点赞
  • 6
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值