tensorflow的Convolution的实现在哪里答案

【问题标题】：Where is the implementation of Convolution of tensorflowtensorflow的Convolution的实现在哪里
【发布时间】：2017-04-18 01:17:41
【问题描述】：

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

【问题讨论】：

标签： tensorflow deep-learning

【解决方案1】：

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

【讨论】：