Tensorflow如何实现其卷积运算？我知道它是基于Eigen库的，我将它追溯到/tensorflow/core/kernels/conv_ops.cc中的 LaunchGeneric.launch ，但它在哪里？我猜是在/third_party_eigen3/unsupported/Eigen/CXX11/src/NeuralNetworks/SpatialConvolutions.h，但是我检查了eigen3下的BUILD文件，似乎没有包含文件来编译库 .
LaunchGeneric.launch
GPU实现只是cuDNN . 我认为你're right on the CPU implementation, and it'包含在eigen's BUILD file中通过 "Eigen/**" （递归通配符） .
"Eigen/**"
1 回答
GPU实现只是cuDNN . 我认为你're right on the CPU implementation, and it'包含在eigen's BUILD file中通过
"Eigen/**"（递归通配符） .