d1e6e624ef891543c5bba32bb0a696d350714693,coremltools/converters/nnssa/coreml/ssa_converter.py,SSAConverter,_convert_reduction,#SSAConverter#Any#,1424

Before Change

    def _convert_reduction(self, node):
        assert len(node.inputs) == 2
        input_name = self._get_input_tensors(node)[0]
        reduction_indices = node.attr.get("reduction_indices")
        keepdims = node.attr.get("keep_dims")
        op = node.op.lower()  // type of the unary operator
        if op in ["all", "any"]:
            op = "prod" if op == "all" else "sum"

After Change

        else:
            reduction_indices = node.attr["axis"]

        if "keep_dims" in node.attr:
            keepdims = node.attr["keep_dims"]
        else:
            keepdims = node.attr["keepdims"]

        op = node.op.lower()  // type of the unary operator
        if op in ["all", "any"]:
            op = "prod" if op == "all" else "sum"

Instances