TensorFlow
文章平均质量分 53
ACodeDog
小码农一只
展开
-
tensorflow c++算子快速实践
背景在c++算子优化的过程中,开发完的算子需要怼进TensorFlow的一套流程中(c++端开发算子,c++端和python对对接,python端封装等)。上述流程比较复杂。用以下办法,可以快速实践c++算子的效果。编写c++算子按照TensorFlow c++ kernels的规则编写算子。将c++算子打包成.so文件TF_CFLAGS=( $(python -c 'import tensorflow as tf; print(" ".join(tf.sysconfig.get_原创 2021-09-03 15:24:56 · 478 阅读 · 0 评论 -
Tensorflow dataset
name 用法 usage 大致用法解释 types intro TextLineDataset tf.data.TextLineDataset 读取文件Dataset 从text文件中读取一行数据 可选,默认256k 无辅线程 buffer_size为配置buffer大小、 ParallelInterleaveDataset tf.dat...原创 2021-01-10 15:05:39 · 550 阅读 · 0 评论 -
Tensorflow错误码一览表
IDNAMETEXT大意0OK成功1CANCELLEDThe operation was cancelled (typically by the caller).被取消2UNKNOWNUnknown error. An example of where this error may be returned is if a Status value received from another address space belongs to an erro...原创 2021-01-10 14:24:22 · 567 阅读 · 0 评论 -
Tensorflow C++端打印tensor,python端打印feature
std::cout << "dtype:" << DataTypeString(x.dtype()) << std::endl; // 输出tensor类型std::cout << "nums:" << x.NumElements() << std::endl; // 输出tensor 包涵的元素个数for (uint32_t i = 0; i < x.NumElements(); ++i) { std::cout &原创 2021-01-10 14:06:30 · 1140 阅读 · 0 评论 -
TensorFlow编译常见问题
swig找不到c++abi--- tensorflow/tensorflow.bzl.orig 2017-11-01 21:21:13.000000000 +0100+++ tensorflow/tensorflow.bzl 2017-11-09 11:49:41.536361562 +0100@@ -920,6 +920,7 @@ args += [src.path] outp...原创 2020-04-27 16:01:35 · 253 阅读 · 0 评论 -
Tensorflow源码分析之ExecutorState
写在前面该文章以尽量详细的注释来表明源代码中每一句的含义。Tensorflow版本:1.10我会以自己习惯的顺序或者便于理解的顺序或者仅我知道的地方来说,如有错误,请评论指出,感激不尽~.该代码注释仅作为记录,不作为其他使用1. ScheduleReady //该函数主要将该次的Process函数处理完成过后得到的ready队列里的node分配到各个线程中去 void Execu...原创 2020-04-21 23:32:23 · 635 阅读 · 0 评论 -
bazel跑TensorFlow中c++测试用例(test)
跟Python的有所不同,TensorFlow中c++测试用例不是写在bazel中的。而是以一定规则写的。例如,如下:bazel test :lib_strings_scanner_testbazel test :哪一级目录下的哪个test...原创 2020-03-18 16:06:38 · 804 阅读 · 0 评论 -
tensorflow dataset模块
写在前面:Python 3.6.5-debugApple LLVM version 10.0.1 (clang-1001.0.46.4)Tensorflow 1.10.1-debug1.Graph构图Graph构图过程,顾名思义,主要讲session run函数调用之前,将图中的每个节点都构建入graph内执行步骤。Dataset 分为两种,1. 真正获取数据的datase...原创 2019-11-16 15:42:16 · 1902 阅读 · 0 评论 -
Tensorflow快速编译
1. 配置环境变量使得./configure不用手动选择是否下载全新的llvm等,环境变量如下:export TF_NEED_SEASTAR=0#还有很多类似的,详见configure.py2. 在TensorFlow的编译目录里执行:tensorflow/tools/ci_build/builds/configured CPU来reconfig。3. 使用configu...原创 2020-02-27 23:02:33 · 579 阅读 · 0 评论