experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

tf.depth_to_space is now tf.compat.v1.depth_to_space

Browse Source 
PiperOrigin-RevId: 276387363
Change-Id: If72ea522ca1b7c2154e8e1258604afff99e8c917
This commit is contained in:
Brian Atkinson 2019-10-23 17:39:54 -07:00 committed by TensorFlower Gardener
parent f2e7824b78
commit 79230ddda7
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
tensorflow/lite/testing/op_tests/depth_to_space.py
View File

@ -38,7 +38,8 @@ def make_depth_to_space_tests(options):
dtype=parameters["dtype"], dtype=parameters["dtype"],
name="input", name="input",
shape=parameters["input_shape"]) 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] return [input_tensor], [out]
def build_inputs(parameters, sess, inputs, outputs): def build_inputs(parameters, sess, inputs, outputs):