diff --git a/ci_build/azure_pipelines/onnxruntime_nightly_test.yml b/ci_build/azure_pipelines/onnxruntime_nightly_test.yml index 06b2de99e..9cd22c115 100644 --- a/ci_build/azure_pipelines/onnxruntime_nightly_test.yml +++ b/ci_build/azure_pipelines/onnxruntime_nightly_test.yml @@ -80,7 +80,7 @@ stages: parameters: platforms: ['linux'] python_versions: ['3.9'] - tf_versions: ['2.6.0'] + tf_versions: ['2.6.1'] onnx_opsets: [''] onnx_backends: {onnxruntime: ['nightly']} job: diff --git a/ci_build/azure_pipelines/pretrained_model_test-matrix.yml b/ci_build/azure_pipelines/pretrained_model_test-matrix.yml index 17740bd87..9c3cc3cbd 100755 --- a/ci_build/azure_pipelines/pretrained_model_test-matrix.yml +++ b/ci_build/azure_pipelines/pretrained_model_test-matrix.yml @@ -41,7 +41,7 @@ jobs: parameters: platforms: ['linux', 'windows'] python_versions: ['3.9'] - tf_versions: ['2.6.0'] + tf_versions: ['2.6.1'] job: steps: - template: 'pretrained_model_test.yml' diff --git a/ci_build/azure_pipelines/pretrained_model_test.yml b/ci_build/azure_pipelines/pretrained_model_test.yml index d8e356a3c..0c552b8f8 100644 --- a/ci_build/azure_pipelines/pretrained_model_test.yml +++ b/ci_build/azure_pipelines/pretrained_model_test.yml @@ -15,7 +15,7 @@ jobs: - template: 'templates/job_generator.yml' parameters: python_versions: ['3.7'] - tf_versions: ['1.15.5','2.6.0'] + tf_versions: ['1.15.5','2.6.1'] job: steps: - template: 'pretrained_model_test.yml' diff --git a/ci_build/azure_pipelines/unit_test-matrix.yml b/ci_build/azure_pipelines/unit_test-matrix.yml index 1b300562a..d8e138178 100644 --- a/ci_build/azure_pipelines/unit_test-matrix.yml +++ b/ci_build/azure_pipelines/unit_test-matrix.yml @@ -51,7 +51,7 @@ stages: parameters: platforms: ['linux', 'windows'] python_versions: ['3.8'] - tf_versions: ['2.6.0'] + tf_versions: ['2.6.1'] onnx_opsets: [''] job: steps: diff --git a/ci_build/azure_pipelines/unit_test.yml b/ci_build/azure_pipelines/unit_test.yml index 662eb3ed2..3faba5d4e 100644 --- a/ci_build/azure_pipelines/unit_test.yml +++ b/ci_build/azure_pipelines/unit_test.yml @@ -7,7 +7,7 @@ stages: parameters: # TFJS tf 2.6 python_versions: ['3.9'] - tf_versions: ['2.6.0'] + tf_versions: ['2.6.1'] onnx_opsets: [''] skip_tfjs_tests: 'False' skip_tf_tests: 'True' @@ -20,7 +20,7 @@ stages: parameters: # TFLite tf 2.6 python_versions: ['3.8'] - tf_versions: ['2.6.0'] + tf_versions: ['2.6.1'] onnx_opsets: [''] skip_tflite_tests: 'False' skip_tf_tests: 'True' @@ -33,7 +33,7 @@ stages: parameters: # tf 2.6 python_versions: ['3.8'] - tf_versions: ['2.6.0'] + tf_versions: ['2.6.1'] onnx_opsets: [''] job: steps: diff --git a/tests/test_backend.py b/tests/test_backend.py index 512350be2..1fd7d3ce6 100644 --- a/tests/test_backend.py +++ b/tests/test_backend.py @@ -2681,6 +2681,7 @@ def func(x, y, z): self._run_test_case(func, [_OUTPUT], {_INPUT: x_val, _INPUT1: y_val, _INPUT2: z_val}) @check_opset_min_version(10, "Slice") + @skip_tfjs("TFJS executes model incorrectly") def test_new_axis_mask(self): def func(x, y): x_ = x[tf.newaxis, 0:y, y::2, tf.newaxis, :, tf.newaxis, :y, tf.newaxis, ..., 9]