tf.depth_to_space is now tf.compat.v1.depth_to_space
PiperOrigin-RevId: 276387363 Change-Id: If72ea522ca1b7c2154e8e1258604afff99e8c917
This commit is contained in:
parent
f2e7824b78
commit
79230ddda7
@ -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):
|
||||
|
Loading…
Reference in New Issue
Block a user