diff --git a/testdata/dnn/tensorflow/generate_tf_models.py b/testdata/dnn/tensorflow/generate_tf_models.py index 0543a97cb..8c282228d 100644 --- a/testdata/dnn/tensorflow/generate_tf_models.py +++ b/testdata/dnn/tensorflow/generate_tf_models.py @@ -672,6 +672,11 @@ def pad_depth(x, desired_channels): sub = conv - inp save(inp, sub, 'eltwise_sub') ################################################################################ +inp = tf.placeholder(tf.float32, [None, 2, 3, 4], 'input') +conv = tf.layers.conv2d(inp, filters=3, kernel_size=[1, 1]) +softmax = tf.contrib.slim.softmax(conv) +save(inp, softmax, 'slim_softmax') +################################################################################ # Uncomment to print the final graph. # with tf.gfile.FastGFile('fused_batch_norm_net.pb') as f: diff --git a/testdata/dnn/tensorflow/slim_softmax_in.npy b/testdata/dnn/tensorflow/slim_softmax_in.npy new file mode 100644 index 000000000..63d6f8aab Binary files /dev/null and b/testdata/dnn/tensorflow/slim_softmax_in.npy differ diff --git a/testdata/dnn/tensorflow/slim_softmax_net.pb b/testdata/dnn/tensorflow/slim_softmax_net.pb new file mode 100644 index 000000000..f888e8fa6 Binary files /dev/null and b/testdata/dnn/tensorflow/slim_softmax_net.pb differ diff --git a/testdata/dnn/tensorflow/slim_softmax_out.npy b/testdata/dnn/tensorflow/slim_softmax_out.npy new file mode 100644 index 000000000..d2e7cb5b1 Binary files /dev/null and b/testdata/dnn/tensorflow/slim_softmax_out.npy differ