From 79230ddda7b607513e9160129ab50bd685b93d88 Mon Sep 17 00:00:00 2001 From: Brian Atkinson Date: Wed, 23 Oct 2019 17:39:54 -0700 Subject: [PATCH] tf.depth_to_space is now tf.compat.v1.depth_to_space PiperOrigin-RevId: 276387363 Change-Id: If72ea522ca1b7c2154e8e1258604afff99e8c917 --- tensorflow/lite/testing/op_tests/depth_to_space.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/tensorflow/lite/testing/op_tests/depth_to_space.py b/tensorflow/lite/testing/op_tests/depth_to_space.py index faf7f64826f..13859017a59 100644 --- a/tensorflow/lite/testing/op_tests/depth_to_space.py +++ b/tensorflow/lite/testing/op_tests/depth_to_space.py @@ -38,7 +38,8 @@ def make_depth_to_space_tests(options): dtype=parameters["dtype"], name="input", shape=parameters["input_shape"]) - out = tf.depth_to_space(input_tensor, block_size=parameters["block_size"]) + out = tf.compat.v1.depth_to_space( + input_tensor, block_size=parameters["block_size"]) return [input_tensor], [out] def build_inputs(parameters, sess, inputs, outputs):