fix bug

if as_list is set to be true, then get_tensor_value will remain the dtype
info; while if it's false then dtype info is removed. this will trigger a
bug sometimes

Author: zhijxu-MS

tf2onnx/graph_builder.py

@@ -70,6 +70,11 @@ def make_slice(self, kwargs, name=None, shapes=None, dtypes=None):
         while inputs[-1] == "":
             inputs = inputs[:-1]
 
+        if self.graph.opset >= 10:
+            dtype = self.graph.get_dtype(inputs[1])
+            for input_data in inputs[1:]:
+                utils.make_sure(dtype == self.graph.get_dtype(input_data), "dtype should be same")
+
         return self.graph.make_node(op_type="Slice", inputs=inputs, attr=attr, name=name,
                                     outputs=outputs, shapes=shapes, dtypes=dtypes).output[0]
 

tf2onnx/onnx_opset/tensor.py

@@ -532,11 +532,12 @@ def version_4(cls, ctx, node, **kwargs):
             attr = node.get_attr(attr_name)
             if attr is not None and attr.i != 0:
                 raise ValueError("StridedSlice: attribute " + attr_name + " not supported")
-        input_shape = ctx.get_shape(node.input[0])
-        begin = node.inputs[1].get_tensor_value(as_list=False)
-        end = node.inputs[2].get_tensor_value(as_list=False)
-        strides = node.inputs[3].get_tensor_value(as_list=False)
-        max_size = np.iinfo(begin.dtype).max
+        onnx_dtype = ctx.get_dtype(node.input[1])
+        np_dtype = utils.ONNX_TO_NUMPY_DTYPE[onnx_dtype]
+        max_size = np.iinfo(np_dtype).max
+        begin = node.inputs[1].get_tensor_value()
+        end = node.inputs[2].get_tensor_value()
+        strides = node.inputs[3].get_tensor_value()
         end_mask = node.get_attr("end_mask")
         end_mask = end_mask.i if end_mask is not None else 0
         begin_mask = node.get_attr("begin_mask")