Update end2end_tfkeras.py (#2361)

Tensorflow Keras save model function change to export for `pb` export extension

Signed-off-by: Masoud Kaviani <kaviani.masoud@gmail.com>
Co-authored-by: Jay Zhang <36183870+fatcat-z@users.noreply.github.com>

Author: MasoudKaviani

examples/end2end_tfkeras.py

@@ -41,9 +41,7 @@
 
 ########################################
 # Saves the model.
-if not os.path.exists("simple_rnn"):
-    os.mkdir("simple_rnn")
-tf.keras.models.save_model(model, "simple_rnn")
+model.export("simple_rnn")
 
 ########################################
 # Run the command line.
@@ -57,7 +55,7 @@
 ########################################
 # Runs onnxruntime.
 session = InferenceSession("simple_rnn.onnx")
-got = session.run(None, {'input_1': input})
+got = session.run(None, {'keras_tensor': input})
 print(got[0])
 
 ########################################
@@ -68,5 +66,5 @@
 # Measures processing time.
 print('tf:', timeit.timeit('model.predict(input)',
                            number=100, globals=globals()))
-print('ort:', timeit.timeit("session.run(None, {'input_1': input})",
+print('ort:', timeit.timeit("session.run(None, {'keras_tensor': input})",
                             number=100, globals=globals()))