test tf1

onnx · May 21, 2021 · 6b5a797 · 6b5a797
1 parent 75d09e1
commit 6b5a797
Show file tree

Hide file tree

Showing 5 changed files with 7 additions and 6 deletions.
diff --git a/.azure-pipelines/linux-conda-CI-tf-keras.yml b/.azure-pipelines/linux-conda-CI-tf-keras.yml
@@ -66,7 +66,7 @@ jobs:
       pip install -r requirements.txt
       pip install -r requirements-dev.txt
       $(INSTALL_ORT)
-      pip install git+https://github.com/onnx/tensorflow-onnx
+      pip install git+https://github.com/onnx/tensorflow-onnx@tom/test_keras_tf1
     displayName: 'Install dependencies'
 
   - script: |
@@ -187,7 +187,7 @@ jobs:
       pip install git+https://github.com/qubvel/efficientnet
       $(INSTALL_TRANSFORMERS)
       pip install keras-self-attention
-      pip install git+https://github.com/onnx/tensorflow-onnx
+      pip install git+https://github.com/onnx/tensorflow-onnx@tom/test_keras_tf1
     displayName: 'Install dependencies'
 
   - script: |

diff --git a/.azure-pipelines/linux-conda-CI.yml b/.azure-pipelines/linux-conda-CI.yml
@@ -73,7 +73,7 @@ jobs:
       pip install -r requirements.txt
       pip install -r requirements-dev.txt
       $(INSTALL_ORT)
-      pip install git+https://github.com/onnx/tensorflow-onnx
+      pip install git+https://github.com/onnx/tensorflow-onnx@tom/test_keras_tf1
     displayName: 'Install dependencies'
 
   - script: |

diff --git a/.azure-pipelines/win32-conda-CI-tf-keras.yml b/.azure-pipelines/win32-conda-CI-tf-keras.yml
@@ -69,7 +69,7 @@ jobs:
       pip install -r requirements.txt
       pip install -r requirements-dev.txt
       %INSTALL_ORT%
-      pip install git+https://github.com/onnx/tensorflow-onnx
+      pip install git+https://github.com/onnx/tensorflow-onnx@tom/test_keras_tf1
     displayName: 'Install dependencies'
 
   - script: |
@@ -186,7 +186,7 @@ jobs:
       pip install git+https://github.com/qubvel/efficientnet
       %INSTALL_TRANSFORMERS%
       pip install keras-self-attention
-      pip install git+https://github.com/onnx/tensorflow-onnx
+      pip install git+https://github.com/onnx/tensorflow-onnx@tom/test_keras_tf1
     displayName: 'Install dependencies'
 
   - script: |

diff --git a/.azure-pipelines/win32-conda-CI.yml b/.azure-pipelines/win32-conda-CI.yml
@@ -69,7 +69,7 @@ jobs:
       pip install -r requirements.txt
       pip install -r requirements-dev.txt
       %INSTALL_ORT%
-      pip install git+https://github.com/onnx/tensorflow-onnx
+      pip install git+https://github.com/onnx/tensorflow-onnx@tom/test_keras_tf1
     displayName: 'Install dependencies'
 
   - script: |

diff --git a/keras2onnx/main.py b/keras2onnx/main.py
@@ -43,6 +43,7 @@ def convert_keras_tf2onnx(model, name=None, doc_string='', target_opset=None, in
         target_opset = get_maximum_opset_supported()
     input_signature = _process_initial_types(initial_types, unknown_dim=None)
 
+
     import tf2onnx
     model, external_tensor_storage = tf2onnx.convert.from_keras(model, input_signature, opset=target_opset,
                                                                 inputs_as_nchw=channel_first_inputs)