From b7687e53d1c50b7be49ca2479af465926005be00 Mon Sep 17 00:00:00 2001 From: Yanhui Liang Date: Mon, 9 Nov 2020 15:20:48 -0800 Subject: [PATCH] Replace the usages of `array_ops.pack` with its TF public api. PiperOrigin-RevId: 341496338 Change-Id: If79be29825fbb67d53be04a37505661df108ac62 --- tensorflow/python/keras/tests/tracking_test.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/tensorflow/python/keras/tests/tracking_test.py b/tensorflow/python/keras/tests/tracking_test.py index f3818190902..0e1384f52d8 100644 --- a/tensorflow/python/keras/tests/tracking_test.py +++ b/tensorflow/python/keras/tests/tracking_test.py @@ -34,6 +34,7 @@ from tensorflow.python.keras.layers import core from tensorflow.python.keras.layers import normalization from tensorflow.python.module import module from tensorflow.python.ops import array_ops +from tensorflow.python.ops import gen_array_ops from tensorflow.python.ops import math_ops from tensorflow.python.ops import variables from tensorflow.python.platform import test @@ -226,7 +227,7 @@ class ListTests(keras_parameterized.TestCase): self.assertAllEqual( [1., 2., 3.], - self.evaluate(array_ops.pack(ListToTensor().l))) + self.evaluate(gen_array_ops.Pack(values=ListToTensor().l))) class ListWrapperTest(test.TestCase): @@ -540,7 +541,7 @@ class TupleTests(keras_parameterized.TestCase): self.assertAllEqual( (1., 2., 3.), - self.evaluate(array_ops.pack(TupleToTensor().l))) + self.evaluate(gen_array_ops.Pack(values=TupleToTensor().l))) class InterfaceTests(keras_parameterized.TestCase):