fixed simple FP16 test

2019-08-06 17:14:16 -07:00
parent 6c39cdbace
commit 0e201e18ff
5 changed files with 33 additions and 21 deletions
--- a/examples/python/tensorflow/run.py
+++ b/examples/python/tensorflow/run.py
@@ -23,7 +23,7 @@ def run_dot():
    result = sess.run([c], feed_dict = {a: ha,
                                        b: hb})[0]
    # Test
-    hresult = np.dot(ha.T, hb).T
+    hresult = np.dot(ha.T, hb.T).T
    dif = np.abs(result - hresult)
    np.savetxt('dif.dat', dif, '%2.4f')
    print(hresult)