C++ tensorflow内核代码修改
我想测量每个函数花费的时间。 所以我修改了tensorflow/core/kernel/conv_ops.cc,如下所示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
....
#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,看看发生了什么