我可以看到它在 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,志愿者(:实现下面的内核。