Können Sie den QR-Faktorisierungsalgorithmus von cuSOLVER im Tensorflow verwenden?

Question

Ich kann sehen, es ist in https://github.com/tensorflow/tensorflow/blob/master/tensorflow/core/kernels/cuda_solvers.h (Geqrf) referenziert, aber die Tensorflow QR-Op bei https://github.com/tensorflow/tensorflow/blob/master/tensorflow/core/kernels/qr_op_float.cc ist nicht für die GPU-Verwendung registriert. Kann ich irgendwie mit Geqrf meine eigene Op machen, oder gibt es vielleicht einen anderen Weg?

Answer 1

This line im Quellcode, den Sie verknüpft gibt einen Hinweis auf die aktuelle Situation.

TODO (rmlaren, Freiwillige): Implementieren Sie die folgenden Kernel.