2b7f4471e764dd4b1a924837b6f515ad7e8a294a,coremltools/converters/nnssa/coreml/ssa_converter.py,SSAConverter,_convert_slice,#SSAConverter#Any#,666

Before Change

        if "slice" not in node.attr:
            assert node.attr["new_axis_mask"] == 0
            assert len(input_names) >= 4
            rank = len(self._get_tensor_shape_from_type(input_nodes[].datatype))
            begin_masks = [True if i in node.attr["begin_masks"] else False for i in range(rank)]
            end_masks = [True if i in node.attr["end_masks"] else False for i in range(rank)]
            layer = builder.add_slice_dynamic(name=slice_output_name,

After Change

            shapes.propagate_single_layer(layer, self.tensor_shapes)

        if has_squeeze:
            input_shape = self._get_tensor_shape_from_type(input_types[0])
            input_rank  = len(input_shape)
            squeeze_all = (input_rank == len(axes))
            layer = builder.add_squeeze(
                name=node.name,

Instances