我是tensorflow源代码修改的新手。
我想尝试一种conv2d算法的变体，想知道源代码中输入/内核matmul的确切位置，以及如何通过修改重建tensorflow。
答案 0 :(得分：0)
您可以阅读有关如何向TensorFlow添加新操作的this教程。您将学习如何找到ops的实现（即那里的内核）。在github页面上有关于从源编译tensorflow的明确说明。
对于Conv2D，操作在core / ops / nn_ops.cc中定义。它有许多内核（用于CPU，用于GPU，用于XLA，基于MKL，基于特征）。如果您这样做，可以看到它们：tensorflow/core/kernels$ ls | grep conv