Skip to content

Commit af56985

Browse files
committed
Merge pull request #893 from rogday:efficientnet_fix_34
2 parents 87f2521 + a43fbb5 commit af56985

12 files changed

+16
-0
lines changed

testdata/dnn/tensorflow/generate_tf_models.py

Lines changed: 16 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -572,6 +572,22 @@ def keras_relu6(x):
572572
save(inp, reduced, 'reduce_mean')
573573
################################################################################
574574
inp = tf.placeholder(tf.float32, [2, 3, 4, 5], 'input')
575+
reduced = tf.reduce_max(inp, axis=[1, 2], keepdims=True)
576+
save(inp, reduced, 'reduce_max')
577+
################################################################################
578+
inp = tf.placeholder(tf.float32, [1, 2, 2, 4], 'ReduceMax')
579+
out = tf.reduce_max([inp, inp * 2], axis=0)
580+
save(inp, out, 'max_pool_by_axis')
581+
################################################################################
582+
inp = tf.placeholder(tf.float32, [1, 4, 2, 3], 'input')
583+
out = tf.math.reduce_max(inp, axis=-1)
584+
save(inp, out, 'reduce_max_channel')
585+
################################################################################
586+
inp = tf.placeholder(tf.float32, [1, 4, 2, 3], 'input')
587+
out = tf.math.reduce_max(inp, axis=-1, keep_dims=True)
588+
save(inp, out, ('reduce_max_channel', 'keep_dims'), is_gen_data=False)
589+
################################################################################
590+
inp = tf.placeholder(tf.float32, [2, 3, 4, 5], 'input')
575591
reduced = tf.reduce_sum(inp, axis=[1, 2], keepdims=True)
576592
save(inp, reduced, 'reduce_sum')
577593
################################################################################
192 Bytes
Binary file not shown.
403 Bytes
Binary file not shown.
192 Bytes
Binary file not shown.
224 Bytes
Binary file not shown.
Binary file not shown.
Binary file not shown.
232 Bytes
Binary file not shown.
160 Bytes
Binary file not shown.
608 Bytes
Binary file not shown.
234 Bytes
Binary file not shown.
168 Bytes
Binary file not shown.

0 commit comments

Comments
 (0)