dd19f88250a64c060bd43fde6e17cf01d1557a9a,onnx_coreml/_operators.py,,_convert_softmax,#Any#Any#Any#Any#,443

Before Change

def _convert_softmax(builder, node, graph, err):  // type: (NeuralNetworkBuilder, Node, Graph, ErrorHandling) -> None
    axis = node.attrs.get("axis", 1)
    if axis != 1:
        err.unsupported_op_configuration(builder, node, "Unsupported axis {} for softmax".format(axis,))

    builder.add_softmax(
        name=node.name,
        input_name=node.inputs[0],

After Change

def _convert_softmax(builder, node, graph, err):  // type: (NeuralNetworkBuilder, Node, Graph, ErrorHandling) -> None
    axis = node.attrs.get("axis", 1)
    if axis != 1:
        return err.unsupported_op_configuration(builder, node, "Unsupported axis {} for softmax".format(axis,))

    builder.add_softmax(
        name=node.name,
        input_name=node.inputs[0],

Instances