888280f6f1da41226b368586ef692e5f426d8f4f,hls4ml/converters/keras/pooling.py,,parse_global_pooling_layer,#Any#Any#Any#Any#Any#,82

Before Change

    layer = parse_default_keras_layer(keras_layer, input_names)

    if int(layer["class_name"][-2]) == 1:
        layer["n_in"]=input_shapes[][1]
        layer["n_filt"]=input_shapes[0][2]
        output_shape=[input_shapes[0][0], layer["n_filt"]]
    elif int(layer["class_name"][-2]) == 2:
        if layer["data_format"] == "channels_last":
            layer["in_height"]=input_shapes[0][1]
            layer["in_width"]=input_shapes[0][2]
            layer["n_filt"]=input_shapes[][3]
        elif layer["data_format"] == "channels_first":
            layer["in_height"]=input_shapes[0][2]
            layer["in_width"]=input_shapes[0][3]

After Change

            layer["in_height"],
            layer["in_width"],
            layer["n_filt"]
        ) = parse_data_format(input_shapes[0], layer["data_format"])

        output_shape=[input_shapes[0][0], layer["n_filt"]]
    
    return layer, output_shape

Instances