您可以在张量流中使用 cuSOLVER 中的 GPU QR 分解算法吗?



我可以看到它在 https://github.com/tensorflow/tensorflow/blob/master/tensorflow/core/kernels/cuda_solvers.h(Geqrf(中被引用,但 https://github.com/tensorflow/tensorflow/blob/master/tensorflow/core/kernels/qr_op_float.cc 的张量流QR操作没有注册GPU使用。我可以以某种方式使用 Geqrf 进行自己的操作,还是有其他方法?

您链接的源代码中的这一行提供了有关当前情况的提示。

TODO(rmlarsen,志愿者(:实现下面的内核。

相关内容

  • 没有找到相关文章

最新更新