Warning: file_get_contents(/data/phpspider/zhask/data//catemap/6/cplusplus/150.json): failed to open stream: No such file or directory in /data/phpspider/zhask/libs/function.php on line 167

Warning: Invalid argument supplied for foreach() in /data/phpspider/zhask/libs/tag.function.php on line 1116

Notice: Undefined index: in /data/phpspider/zhask/libs/function.php on line 180

Warning: array_chunk() expects parameter 1 to be array, null given in /data/phpspider/zhask/libs/function.php on line 181
C++ tensorflow内核代码修改_C++_Tensorflow - Fatal编程技术网

C++ tensorflow内核代码修改

C++ tensorflow内核代码修改,c++,tensorflow,C++,Tensorflow,我想测量每个函数花费的时间。 所以我修改了tensorflow/core/kernel/conv_ops.cc,如下所示 .... #include <ctime> .... void Compute(OpKernelContext* context) override { // Input tensor is of the following dimensions: // [ batch, in_rows, in_cols, in_depth ] std::clock_t

我想测量每个函数花费的时间。 所以我修改了tensorflow/core/kernel/conv_ops.cc,如下所示

....
#include <ctime>
....

void Compute(OpKernelContext* context) override {
 // Input tensor is of the following dimensions:
 // [ batch, in_rows, in_cols, in_depth ]
 std::clock_t start;
 double duration;

....
....
....

 duration = (std::clock() - start) / (double) CLOCKS_PER_SEC;
 std::cout<<"============== conv time : "<<duration<<std::endl;
}

....
....
。。。。
#包括
....
无效计算(OpKernelContext*上下文)覆盖{
//输入张量的尺寸如下:
//[批次、行、列、深度]
std::时钟未启动;
双倍持续时间;
....
....
....
持续时间=(标准::时钟()-开始)/(双)每秒时钟;

std::cout使用bazel从源代码重新编译整个tensorflow项目,然后从conv_ops.cc生成gen_nnu ops.py文件。它可能会工作。

使用bazel从源代码重新编译整个tensorflow项目,然后从conv_ops.cc生成gen_nn_ops.py文件。它可能会工作。

您有
\35; include
吗?我已经测试了一个用户定义操作()但在这段代码中,我只添加了“#include”,效果很好。您可能需要重新编译tensorflow?我可以在同一目录中使用“cc-c conv#u ops.cc”吗…?Compile conv#u ops.cc不会将其链接到库。请尝试重新编译tensorflow,看看会发生什么情况。您有一个
#include
?我已经测试了一个用户定义的操作()但在这段代码中,我只添加了“#include”,效果很好。您可能需要重新编译tensorflow?我可以在同一目录中使用“cc-c conv_ops.cc”吗?Compile conv_ops.cc不会将其链接到库。请尝试重新编译tensorflow,看看发生了什么