张量流的卷积的实现在哪里

Question

Tensorflow如何实现其卷积运算？我知道它基于Eigen库，我将它跟踪到/tensorflow/core/kernels/conv_ops.cc中的LaunchGeneric.launch，但它在哪里？我的猜测是在/third_party_eigen3/unsupported/Eigen/CXX11/src/NeuralNetworks/SpatialConvolutions.h，但是我检查了eigen3下的BUILD文件，似乎没有包含文件来编译库。

Answer 1

GPU实现只是cuDNN。我认为你在CPU实现上是正确的，并且它从eigen's BUILD file通过"Eigen/**"（递归通配符）包含在构建中。