def convert_ModulatedConv2d()

in mobile_cv/lut/lib/pt/pt_converter.py [0:0]

• 30 lines of code
• 2 McCabe index (conditional complexity)

def convert_ModulatedConv2d(m, input_shapes):
    dilation = 1
    groups = 1
    bias = None
    output_padding = 0
    if m.upsample:
        op = lut_ops.ConvTranspose2d(
            m.in_channels,
            m.out_channels,
            m.kernel_size,
            m.stride,
            m.padding,
            output_padding,
            groups,
            bias,
            dilation,
        )
    else:
        op = lut_ops.Conv2d(
            m.in_channels,
            m.out_channels,
            m.kernel_size,
            m.stride,
            m.padding,
            dilation,
            groups,
            bias,
        )
    ret = lut_schema.OpInfo(op, input_shapes)
    return ret