Mostly internal changes.
PiperOrigin-RevId: 314275429 Change-Id: I75d2431bee12a4ebc92b5ab291451ecce805f09a
This commit is contained in:
parent
e20f16bfb8
commit
0482d18b52
@ -1017,7 +1017,8 @@ class BinaryOpsTest(xla_test.XLATestCase):
|
||||
math_ops.matmul,
|
||||
np.array([[3.1415926535897932]], dtype=dtype),
|
||||
np.array([[2.7182818284590452]], dtype=dtype),
|
||||
expected=np.array([[8.5397342226735668]], dtype=dtype))
|
||||
expected=np.array([[8.5397342226735668]], dtype=dtype),
|
||||
rtol=1e-14)
|
||||
|
||||
# Edge case with a large range of exponent. Not supported by float16.
|
||||
if dtype != np.float16:
|
||||
@ -1025,7 +1026,8 @@ class BinaryOpsTest(xla_test.XLATestCase):
|
||||
math_ops.matmul,
|
||||
np.array([[9.4039548065783000e-38]], dtype=dtype),
|
||||
np.array([[4.5070591730234615e37]], dtype=dtype),
|
||||
expected=np.array([[4.2384180773686798]], dtype=dtype))
|
||||
expected=np.array([[4.2384180773686798]], dtype=dtype),
|
||||
rtol=1e-14)
|
||||
|
||||
# TODO(phawkins): failing on GPU, no registered kernel.
|
||||
def DISABLED_testSparseMatMul(self):
|
||||
|
Loading…
x
Reference in New Issue
Block a user