tensorflow内核代码修改

Question

我想测量每个功能花费的时间。 所以我修改了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;
}

....
....

保存了这段代码（：wq）后，我运行了一个简单的cnn tensorflow代码。但是我添加的代码不起作用（它没有显示＆＃39; cout＆＃39;结果。）。

如何显示时间结果？

Answer 1

使用bazel从源代码重新编译整个tensorflow项目，然后从conv_ops.cc生成一个gen_nn_ops.py文件。