Je peux le voir est référencé dans https://github.com/tensorflow/tensorflow/blob/master/tensorflow/core/kernels/cuda_solvers.h (Geqrf), mais la tensorflow QR op à https://github.com/tensorflow/tensorflow/blob/master/tensorflow/core/kernels/qr_op_float.cc est pas homologué pour l'utilisation du GPU. Est-ce que je peux faire ma propre opération en utilisant Geqrf, ou y a-t-il un autre moyen?Pouvez-vous utiliser l'algorithme de factorisation QR GPU de cuSOLVER dans tensorflow?
0
A
Répondre
0
This line dans le code source vous donne un indice lié à la situation actuelle.
TODO (rmlarsen, volontaires): Implémentez les noyaux ci-dessous.