張量流卷積的實現在哪裏

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/**"（遞歸通配符）構建。