'QR decomposition of type float16?
https://www.tensorflow.org/api_docs/python/tf/linalg/qr
The documentation states that a tensor must be of type float64, float32, half, complex64, or complex128.
tn = tf.constant([ [1, 2], [3, 4] ], dtype=tf.float32)
[Q, R] = tf.linalg.qr(tn);
tn = tf.constant([ [1, 2], [3, 4] ], dtype=tf.float16)
[Q, R] = tf.linalg.qr(tn);
Using single precision works as intended but half precision returns the following error:
Could not find device for node: {{node Qr}} = Qr[T=DT_HALF, full_matrices=false
Sources
This article follows the attribution requirements of Stack Overflow and is licensed under CC BY-SA 3.0.
Source: Stack Overflow
| Solution | Source |
|---|
