Internal Change

PiperOrigin-RevId: 224875931

Internal Change
PiperOrigin-RevId: 224875931
a404d2ed · A. Unique TensorFlower · TensorFlower Gardener · 2d86af34 · a404d2ed
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

tensorflow/python/framework/test_util.py tensorflow/python/framework/test_util.py +1 -1

未找到文件。
--- a/tensorflow/python/framework/test_util.py
+++ b/tensorflow/python/framework/test_util.py
@@ -1866,7 +1866,7 @@ class TensorFlowTestCase(googletest.TestCase):
    # If a is a tensor then convert it to ndarray
    if isinstance(a, ops.Tensor):
      if isinstance(a, ops._EagerTensorBase):
-        return a.numpy()
+        a = a.numpy()
      else:
        a = self.evaluate(a)
    if not isinstance(a, np.ndarray):