We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent ae4099b commit 3b66fc4Copy full SHA for 3b66fc4
testdata/dnn/tensorflow/generate_tf_models.py
@@ -579,6 +579,14 @@ def keras_relu6(x):
579
reduced = tf.reduce_sum(inp, axis=[2], keepdims=False)
580
save(inp, reduced, 'sum_pool_by_axis')
581
################################################################################
582
+inp = tf.placeholder(tf.float32, [1, 4, 2, 3], 'input')
583
+out = tf.math.reduce_sum(inp, axis=-1)
584
+save(inp, out, 'reduce_sum_channel')
585
+################################################################################
586
587
+out = tf.math.reduce_sum(inp, axis=-1, keep_dims=True)
588
+save(inp, out, ('reduce_sum_channel', 'keep_dims'), is_gen_data=False)
589
590
inp = tf.placeholder(tf.float32, [2, 3, 4, 5], 'input')
591
pool = tf.layers.average_pooling2d(inp, pool_size=1, strides=1, padding='SAME')
592
l2norm = tf.nn.l2_normalize(pool, axis=-1)
testdata/dnn/tensorflow/reduce_sum_channel_in.npy
224 Bytes
testdata/dnn/tensorflow/reduce_sum_channel_keep_dims_net.pb
232 Bytes
testdata/dnn/tensorflow/reduce_sum_channel_keep_dims_out.npy
160 Bytes
testdata/dnn/tensorflow/reduce_sum_channel_net.pb
222 Bytes
testdata/dnn/tensorflow/reduce_sum_channel_out.npy
0 commit comments