张量流卷积的实现在哪里



Tensorflow如何实现其卷积操作?我知道它基于 Eigen 库,我将其追溯到/tensorflow/core/kernels/conv_ops.cc 中的LaunchGeneric.launch,但它去了哪里?我的猜测在/third_party_eigen3/unsupported/Eigen/CXX11/src/NeuralNetworks/SpatialConvolutions.h 中,但我检查了 eigen3 下的 BUILD 文件,似乎该文件不包括在编译库内。

GPU 实现只是 cuDNN。我认为您在 CPU 实现上是对的,它通过 "Eigen/**"(递归通配符(包含在 eigen 的 BUILD 文件中。

相关内容

  • 没有找到相关文章

最新更新