Commit e3413de5 authored by A. Unique TensorFlower's avatar A. Unique TensorFlower Committed by TensorFlower Gardener
Browse files

Add GPU support for self_adjoint_eig a.k.a. tf.linalg.eigh. 

Clean up macros and template specializations in cuda_solvers.cc a bit.

PiperOrigin-RevId: 169715681
parent f927a720
Expand all Show whitespace changes
Inline Side-by-side
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment