c++ - 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 代码。但是我添加的代码不起作用(它不显示“cout”结果。)。

如何显示计时结果？

Answer 1

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