当前位置:首页 » 《关注互联网》 » 正文

Tensorflow 源码算子实现分析_tugouxp的专栏

7 人参与  2021年08月20日 10:03  分类 : 《关注互联网》  评论

点击全文阅读


本文参考https://www.tensorflow.org/install/source

下载Tensorflow源码:

git clone https://github.com/tensorflow/tensorflow.git

下载完成后,可以看到,tensorflow目录结构如下:

maliankang@AwExdroid112:~/tensorflow$ tree -L 2 -d
.
├── tensorflow
│   ├── c
│   ├── cc
│   ├── compiler
│   ├── core
│   ├── distribute
│   ├── docs_src
│   ├── examples
│   ├── go
│   ├── java
│   ├── js
│   ├── lite
│   ├── python
│   ├── security
│   ├── stream_executor
│   └── tools
├── third_party
│   ├── absl
│   ├── android
│   ├── aws
│   ├── benchmark
│   ├── boringssl
│   ├── clang_toolchain
│   ├── clog
│   ├── compute_library
│   ├── cpuinfo
│   ├── dlpack
│   ├── eigen3
│   ├── farmhash
│   ├── fft2d
│   ├── flatbuffers
│   ├── FP16
│   ├── gemmlowp
│   ├── git
│   ├── googleapis
│   ├── gpus
│   ├── grpc
│   ├── hadoop
│   ├── hexagon
│   ├── highwayhash
│   ├── hwloc
│   ├── icu
│   ├── jpeg
│   ├── kissfft
│   ├── llvm
│   ├── llvm_openmp
│   ├── mkl
│   ├── mkl_dnn
│   ├── mlir
│   ├── mpi
│   ├── nasm
│   ├── nccl
│   ├── opencl_headers
│   ├── pasta
│   ├── protobuf
│   ├── psimd
│   ├── py
│   ├── python_runtime
│   ├── remote_config
│   ├── ruy
│   ├── sobol_data
│   ├── systemlibs
│   ├── tensorrt
│   ├── tf_runtime
│   └── vulkan_headers
└── tools

66 directories
maliankang@AwExdroid112:~/tensorflow$

算子实现集中分布在tensorflow/tensorflow/lite/kernels目录下:

maliankang@AwExdroid112:~/tensorflow/tensorflow/lite/kernels$ ls -l |grep test|wc -l
139
maliankang@AwExdroid112:~/tensorflow/tensorflow/lite/kernels$ ls -l |wc -l
298
maliankang@AwExdroid112:~/tensorflow/tensorflow/lite/kernels$ 

基本上,每个算子的实现都是一个独立文件,并且对应一个测试文件,基于这个假设(和实际相符),我们可以粗略统计一下算子的个数,就是

文件总数-test文件总数=298-139 = 159,也就是大概有160个算子,是否可以理解为160个算子基本能够满足所有网络的需求呢?我认为是的,基于的考虑是TF是一个比较全面的框架.

以常用的conv算子为例,简单分析一下它的实现:

对应的真正实现在 optimized_ops命名空间中的Conv函数,它在另一个文件中,我们找到它, 万恶的矩阵乘法,一眼能看明白的,肯定是数学系毕业的大佬。


结束!


点击全文阅读


本文链接:http://m.zhangshiyu.com/post/25802.html

算子  文件  大佬  
<< 上一篇 下一篇 >>

  • 评论(0)
  • 赞助本站

◎欢迎参与讨论,请在这里发表您的看法、交流您的观点。

最新文章

  • (番外)+(全书)霍沉洲沈青禾此去经年人未还(霍沉洲沈青禾)_(霍沉洲沈青禾此去经年人未还)列表_笔趣阁(霍沉洲沈青禾)
  • (番外)+(全书)霍沉洲沈青禾(此去经年人未还霍沉洲沈青禾)完结_(霍沉洲沈青禾)列表_笔趣阁(此去经年人未还霍沉洲沈青禾)
  • 「重回八零,拒绝替嫁冲喜」章节彩蛋限时释出‌_卫东玉兰苏夏人气小说未删减节选
  • 重生七零祁同伟不再是农民儿子结局+番外纯净版全书免费重生七零祁同伟不再是农民儿子结局+番外纯净版全书免费
  • 傅雅宁的神女老婆,却在背地承欢作乐顾尘傅雅宁全书在线
  • 全文神女老婆,却在背地承欢作乐全局(顾尘傅雅宁)列表_全文神女老婆,却在背地承欢作乐全局
  • (番外)+(全书)此去经年人未还全书+番外+后续免费下载_(沈青禾霍沉洲)此去经年人未还全书+番外+后续列表_笔趣阁(沈青禾霍沉洲)
  • 完结文毁容的姐姐和瞎眼的我离开后,姜家两兄弟悔哭了+后续列表_完结文毁容的姐姐和瞎眼的我离开后,姜家两兄弟悔哭了+后续(林梦婉)
  • 妻子辱我爸受贿自杀,我掏出一等军功章节选推荐_[陈素云辰朋友]小说精彩章节分享
  • 全书浏览苔藓爬满旧日诺言新上(顾砚廷慕晚夏)_苔藓爬满旧日诺言新上(顾砚廷慕晚夏)全书结局
  • 顾尘傅雅宁(神女老婆,却在背地承欢作乐+后续+结局)结局_(顾尘傅雅宁神女老婆,却在背地承欢作乐+后续+结局全书结局)结局列表_笔趣阁(顾尘傅雅宁)
  • 「老婆怀上助理的孩子后,助理要求我净身出户」章节限时抢先看‌_「黄秋雅秋雅姐刘嘉铭」后续完结版

    关于我们 | 我要投稿 | 免责申明

    Copyright © 2020-2022 ZhangShiYu.com Rights Reserved.豫ICP备2022013469号-1