你可以在 tensorflow 中使用来自 cuSOLVER 的 GPU QR 分解算法吗？答案

【问题标题】：Can you use the GPU QR factorization algorithm from cuSOLVER in tensorflow?你可以在 tensorflow 中使用来自 cuSOLVER 的 GPU QR 分解算法吗？
【发布时间】：2017-12-09 15:26:49
【问题描述】：

我可以看到它在 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 的 tensorflow QR op 未注册用于 GPU 使用。我可以以某种方式使用 Geqrf 制作自己的操作，还是有其他方法？

【问题讨论】：

标签： tensorflow gpu linear-algebra cusolver

【解决方案1】：

您链接的源代码中的This line 提供了有关当前情况的提示。

TODO(rmlarsen，志愿者)：实现下面的内核。

【讨论】：