Handle axes=None for apply_slice when opset < 10 (#50)

* Fixes #17, update clip operator (#18) * Fixes #17, update clip operator (updated in ONNX 11) * handle min or max = None * Handle axes=None for apply_slice when opset < 10 Co-authored-by: Xavier Dupré <[email protected]>
microsoft · Jan 30, 2020 · e42677e · e42677e
1 parent e87c572
commit e42677e
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/onnxconverter_common/onnx_ops.py b/onnxconverter_common/onnx_ops.py
@@ -741,8 +741,12 @@ def apply_slice(scope, input_name, output_name, container, starts, ends,
     name = _create_name_or_use_existing_one(scope, 'Slice', operator_name)
 
     if container.target_opset < 10:
-        container.add_node('Slice', input_name, output_name, name=name,
-                           starts=starts, ends=ends, axes=axes, op_version=1)
+        if axes is None:
+            container.add_node('Slice', input_name, output_name, name=name,
+                               starts=starts, ends=ends, op_version=1)
+        else:
+            container.add_node('Slice', input_name, output_name, name=name,
+                               starts=starts, ends=ends, axes=axes, op_version=1)
     else:
         if container.target_opset == 10:
             op_version = 10