2017-07-05 14 views
0

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 opはGPUの使用に登録されていません。私はどういうわけかGeqrfを使って自分のopを作ることができますか、それとも別の方法がありますか?tensorflowでcuSOLVERのGPU QR分解アルゴリズムを使用できますか?

答えて

0

This lineあなたがリンクしているソースコードでは、現在の状況についてのヒントが得られます。

TODO(rmlarsen、volunteers):下記のカーネルを実装してください。

関連する問題