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

Question

我可以看到它在https://github.com/tensorflow/tensorflow/blob/master/tensorflow/core/kernels/cuda_solvers.h (Geqrf) 中被引用，但是https://github.com/tensorflow上的 tensorflow QR op /tensorflow/blob/master/tensorflow/core/kernels/qr_op_float.cc未注册用于 GPU 使用。我可以使用 Geqrf 以某种方式制作自己的操作，还是有其他方法？

Answer 1

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

TODO（rmlarsen，志愿者）：实现以下内核。