diff --git a/testdata/dnn/tensorflow/generate_tf_models.py b/testdata/dnn/tensorflow/generate_tf_models.py index 1d933d006..8ef276956 100644 --- a/testdata/dnn/tensorflow/generate_tf_models.py +++ b/testdata/dnn/tensorflow/generate_tf_models.py @@ -579,6 +579,14 @@ def keras_relu6(x): reduced = tf.reduce_sum(inp, axis=[2], keepdims=False) save(inp, reduced, 'sum_pool_by_axis') ################################################################################ +inp = tf.placeholder(tf.float32, [1, 4, 2, 3], 'input') +out = tf.math.reduce_sum(inp, axis=-1) +save(inp, out, 'reduce_sum_channel') +################################################################################ +inp = tf.placeholder(tf.float32, [1, 4, 2, 3], 'input') +out = tf.math.reduce_sum(inp, axis=-1, keep_dims=True) +save(inp, out, ('reduce_sum_channel', 'keep_dims'), is_gen_data=False) +################################################################################ inp = tf.placeholder(tf.float32, [2, 3, 4, 5], 'input') pool = tf.layers.average_pooling2d(inp, pool_size=1, strides=1, padding='SAME') l2norm = tf.nn.l2_normalize(pool, axis=-1) diff --git a/testdata/dnn/tensorflow/reduce_sum_channel_in.npy b/testdata/dnn/tensorflow/reduce_sum_channel_in.npy new file mode 100644 index 000000000..e47f17abc Binary files /dev/null and b/testdata/dnn/tensorflow/reduce_sum_channel_in.npy differ diff --git a/testdata/dnn/tensorflow/reduce_sum_channel_keep_dims_net.pb b/testdata/dnn/tensorflow/reduce_sum_channel_keep_dims_net.pb new file mode 100644 index 000000000..e34ce71d1 Binary files /dev/null and b/testdata/dnn/tensorflow/reduce_sum_channel_keep_dims_net.pb differ diff --git a/testdata/dnn/tensorflow/reduce_sum_channel_keep_dims_out.npy b/testdata/dnn/tensorflow/reduce_sum_channel_keep_dims_out.npy new file mode 100644 index 000000000..a933456ee Binary files /dev/null and b/testdata/dnn/tensorflow/reduce_sum_channel_keep_dims_out.npy differ diff --git a/testdata/dnn/tensorflow/reduce_sum_channel_net.pb b/testdata/dnn/tensorflow/reduce_sum_channel_net.pb new file mode 100644 index 000000000..83636796b Binary files /dev/null and b/testdata/dnn/tensorflow/reduce_sum_channel_net.pb differ diff --git a/testdata/dnn/tensorflow/reduce_sum_channel_out.npy b/testdata/dnn/tensorflow/reduce_sum_channel_out.npy new file mode 100644 index 000000000..62646e1ec Binary files /dev/null and b/testdata/dnn/tensorflow/reduce_sum_channel_out.npy differ