2017-07-05 1 views
0

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?

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.