diff --git a/docker/install/ubuntu_install_tensorflow.sh b/docker/install/ubuntu_install_tensorflow.sh index 25543909d78b..4e766d4d5a5b 100755 --- a/docker/install/ubuntu_install_tensorflow.sh +++ b/docker/install/ubuntu_install_tensorflow.sh @@ -20,4 +20,4 @@ set -e set -u set -o pipefail -pip3 install tensorflow==2.1.0 keras==2.3.1 h5py +pip3 install tensorflow==2.3.1 keras==2.3.1 h5py diff --git a/python/tvm/relay/frontend/tensorflow.py b/python/tvm/relay/frontend/tensorflow.py index 3df582a0c76a..b0aefe9ed375 100644 --- a/python/tvm/relay/frontend/tensorflow.py +++ b/python/tvm/relay/frontend/tensorflow.py @@ -1334,7 +1334,7 @@ def _impl(inputs, attr, params, mod): op_name="batch_norm", transforms={"scale_after_normalization": "scale", "variance_epsilon": "epsilon"}, extras={"axis": axis}, - ignores=["data_format", "U"], + ignores=["data_format", "U", "exponential_avg_factor"], disables=["momentum"], )(inputs, attr) @@ -1364,7 +1364,7 @@ def _impl(inputs, attr, params, mod): op_name="batch_norm", transforms={"scale_after_normalization": "scale", "variance_epsilon": "epsilon"}, extras={"axis": axis}, - ignores=["data_format"], + ignores=["data_format", "exponential_avg_factor"], disables=["momentum"], )(new_inputs, attr) diff --git a/tests/python/frontend/tensorflow/test_forward.py b/tests/python/frontend/tensorflow/test_forward.py index 8e347e754b98..0bcac7fb1a21 100644 --- a/tests/python/frontend/tensorflow/test_forward.py +++ b/tests/python/frontend/tensorflow/test_forward.py @@ -3644,7 +3644,7 @@ def _test_math_op(op, dtypes=["int32", "float32"]): _test_math_op(tf.math.reduce_max) _test_math_op(tf.math.reduce_min) _test_math_op(tf.math.reduce_prod) - _test_math_op(tf.math.reduce_variance) + _test_math_op(tf.math.reduce_variance, dtypes=["float32"]) _test_math_op(tf.math.reduce_std, dtypes=["float32"]) _test_math_op(tf.math.reduce_logsumexp, dtypes=["float32"]) if package_version.parse(tf.VERSION) >= package_version.parse("1.15.0"): @@ -3855,11 +3855,11 @@ def test_forward_unravel_index(): _test_forward_unravel_index([x, y]) x = np.array([0, 1, 2, 5]) - y = np.array([2, 2]) + y = np.array([2, 3]) _test_forward_unravel_index([x, y]) x = np.array([0, 1, 2, 5]) - y = np.array([2]) + y = np.array([6]) _test_forward_unravel_index([x, y]) x = np.array([102, 300, 16])