Skip to content

Commit c5e6718

Browse files
authored
Merge pull request #940 from pccvlab:MatMul_with_two_inputs
2 parents 934397b + cb19926 commit c5e6718

File tree

7 files changed

+11
-0
lines changed

7 files changed

+11
-0
lines changed
368 Bytes
Binary file not shown.
289 Bytes
Binary file not shown.
320 Bytes
Binary file not shown.

testdata/dnn/tensorflow/generate_tf_models.py

Lines changed: 11 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1055,6 +1055,17 @@ def pad_depth(x, desired_channels):
10551055
expand_dim = tf.expand_dims(expand_dim, 2)
10561056
save(inp, expand_dim, prefix + 'expand_dims_2', optimize=False)
10571057
################################################################################
1058+
inp = tf.placeholder(tf.float32, [2, 3], 'input')
1059+
weights = tf.reshape(inp,[3, 2])
1060+
mm = tf.matmul(inp, weights)
1061+
save(inp, mm, 'two_inputs_matmul')
1062+
################################################################################
1063+
#This test case was generated by tensorflow 1.11.0
1064+
inp = tf.placeholder(tf.float32, [3, 4, 5], 'input')
1065+
weights = tf.reshape(inp,[3, 5, 4])
1066+
bmm = tf.matmul(inp, weights)
1067+
save(inp, bmm, 'batch_matmul', optimize=False)
1068+
#################################################################################
10581069

10591070
# Uncomment to print the final graph.
10601071
# with tf.gfile.FastGFile('fused_batch_norm_net.pb', 'rb') as f:
152 Bytes
Binary file not shown.
255 Bytes
Binary file not shown.
144 Bytes
Binary file not shown.

0 commit comments

Comments
 (0)