テンソルフローのコンボリューションの実装はどこですか

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/**"（再帰的なワイルドカード）経由でビルドに含まれています。