Skip to content

Commit 7c077a7

Browse files
authored
Merge pull request #1077 from dkurt:eltwise_prelu
Test data for PReLU from TensorFlow 2 with non-zero scales
2 parents 9d760cc + da67e69 commit 7c077a7

File tree

4 files changed

+2
-2
lines changed

4 files changed

+2
-2
lines changed

testdata/dnn/tensorflow/generate_tf2_models.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -76,9 +76,9 @@ def saveBroken(graph, name):
7676
save(model, 'tf2_dense', flatten_input=tf.TensorSpec(shape=[None, 1, 2, 3], dtype=tf.float32))
7777
################################################################################
7878
model = tf.keras.models.Sequential([
79-
tf.keras.layers.PReLU(input_shape=(1, 2, 3)),
79+
tf.keras.layers.PReLU(input_shape=(1, 4, 6), alpha_initializer='random_normal'),
8080
])
81-
save(model, 'tf2_prelu', p_re_lu_input=tf.TensorSpec(shape=[None, 1, 2, 3], dtype=tf.float32))
81+
save(model, 'tf2_prelu', p_re_lu_input=tf.TensorSpec(shape=[None, 1, 4, 6], dtype=tf.float32))
8282
################################################################################
8383
model = tf.keras.models.Sequential([
8484
tf.keras.layers.AveragePooling2D(input_shape=(4, 6, 3), pool_size=(2, 2)),
72 Bytes
Binary file not shown.
-231 Bytes
Binary file not shown.
72 Bytes
Binary file not shown.

0 commit comments

Comments
 (0)