您可以在 tensorflow 中使用 cuSOLVER 的 GPU QR 分解算法吗？
Can you use the GPU QR factorization algorithm from cuSOLVER in tensorflow?
我可以看到 https://github.com/tensorflow/tensorflow/blob/master/tensorflow/core/kernels/cuda_solvers.h (Geqrf), but the tensorflow QR op at https://github.com/tensorflow/tensorflow/blob/master/tensorflow/core/kernels/qr_op_float.cc 中引用的它没有注册 GPU 使用。我可以使用 Geqrf 以某种方式制作自己的 op，还是有其他方法？
This line 在您链接的源代码中给出了有关当前情况的提示。
TODO(rmlarsen, volunteers): Implement the kernels below.
我可以看到 https://github.com/tensorflow/tensorflow/blob/master/tensorflow/core/kernels/cuda_solvers.h (Geqrf), but the tensorflow QR op at https://github.com/tensorflow/tensorflow/blob/master/tensorflow/core/kernels/qr_op_float.cc 中引用的它没有注册 GPU 使用。我可以使用 Geqrf 以某种方式制作自己的 op，还是有其他方法？
This line 在您链接的源代码中给出了有关当前情况的提示。
TODO(rmlarsen, volunteers): Implement the kernels below.