diff --git a/Microsoft.ML.TensorFlow.TestModels/model_primitive_input_test/model_primitive_input_test.py b/Microsoft.ML.TensorFlow.TestModels/model_primitive_input_test/model_primitive_input_test.py new file mode 100644 index 0000000..82796f4 --- /dev/null +++ b/Microsoft.ML.TensorFlow.TestModels/model_primitive_input_test/model_primitive_input_test.py @@ -0,0 +1,18 @@ +import tensorflow.compat.v1 as tf + +tf.disable_v2_behavior() + +input1 = tf.placeholder(tf.string, name="input1") +input2 = tf.placeholder(tf.string, name="input2") + +inputs = {'input1': input1, 'input2': input2} + +string_merge = tf.add(input1, input2, name="string_merge") + +outputs = {'string_merge':string_merge} + +with tf.Session() as sess: + string_merge = sess.run(string_merge, feed_dict={input1: "This is Text1", input2: "This is Text2"}) + print (string_merge) + + tf.saved_model.simple_save(sess, r'model_primitive_input_test', inputs=inputs, outputs=outputs ) diff --git a/Microsoft.ML.TensorFlow.TestModels/model_primitive_input_test/saved_model.pb b/Microsoft.ML.TensorFlow.TestModels/model_primitive_input_test/saved_model.pb new file mode 100644 index 0000000..cdf29fc Binary files /dev/null and b/Microsoft.ML.TensorFlow.TestModels/model_primitive_input_test/saved_model.pb differ