Skip to content

Commit 8a5800a

Browse files
authored
Merge pull request #849 from LupusSanctus:am/eltwice_vec
2 parents 71f2370 + 1f99fc6 commit 8a5800a

File tree

4 files changed

+6
-0
lines changed

4 files changed

+6
-0
lines changed
1.1 KB
Binary file not shown.
283 Bytes
Binary file not shown.
Binary file not shown.

testdata/dnn/tensorflow/generate_tf_models.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -786,6 +786,12 @@ def pad_depth(x, desired_channels):
786786
sub = conv - inp
787787
save(inp, sub, 'eltwise_sub')
788788
################################################################################
789+
inp = tf.placeholder(tf.float32, [1, 5, 5, 10], 'input')
790+
out1 = tf.keras.layers.MaxPool2D((2, 2), 5, "SAME", name="pooling")(inp)
791+
out2 = tf.keras.layers.ReLU(name="relu")(inp)
792+
final_out = tf.math.add(out1, out2, name='tf_sum')
793+
save(inp, final_out, 'eltwise_add_vec')
794+
################################################################################
789795
inp = tf.placeholder(tf.float32, [None, 2, 3, 4], 'input')
790796
conv = tf.layers.conv2d(inp, filters=3, kernel_size=[1, 1])
791797
softmax = tf.contrib.slim.softmax(conv)

0 commit comments

Comments
 (0)