7d0d0e6e8b9ef3bb5ed979f6716c84c629eae7a8,src/graph_transpiler/webdnn/graph/converters/keras.py,,_convert_flatten,#Any#Any#,405

Before Change

    :return:
    
    assert len(operator.inputs) == 1
    x = converter.get_variable(operator.inputs[0])
    in_axes = x.order.axes.copy()
    assert in_axes[0] == Axis.N
    in_axes.remove(Axis.N)
    flatten_opr = Flatten(None, in_axes=in_axes, out_axis=Axis.C)
    y, = flatten_opr(x)

    converter.set_variable(operator.get_output_key(0), y)


@KerasConverter.register_handler("Concatenate")
def _convert_concatenate(converter: KerasConverter, operator: KerasOperator):

After Change

    flatten_opr = Flatten(None, in_axes=in_axes, out_axis=Axis.C)
    y, = flatten_opr(x)

    operator.outputs = [y]


@KerasConverter.register_handler("Concatenate")
def _convert_concatenate(converter: KerasConverter, operator: KerasOperator):

Instances