mobile_cv/lut/lib/lut_ops.py [76:106]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
    def __init__(
        self,
        in_channels=-1,
        out_channels=-1,
        kernel_size=-1,
        stride=1,
        padding=0,
        dilation=1,
        groups=1,
        bias=True,
        allow_frac=False,
    ):
        info = {
            "in_channels": in_channels,
            "out_channels": out_channels,
            "kernel_size": kernel_size,
            "stride": stride,
            "padding": padding,
            "dilation": dilation,
            "groups": groups,
            "bias": bias,
        }
        super().__init__(info)
        self._unify()
        self.allow_frac = allow_frac

        if not self.allow_frac:
            in_channels = self.in_channels
            assert in_channels % self.groups == 0, (in_channels, self.groups)

    def get_output_shape(self, input_shape):
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -



mobile_cv/lut/lib/lut_ops.py [154:184]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
    def __init__(
        self,
        in_channels=-1,
        out_channels=-1,
        kernel_size=-1,
        stride=1,
        padding=0,
        dilation=1,
        groups=1,
        bias=True,
        allow_frac=False,
    ):
        info = {
            "in_channels": in_channels,
            "out_channels": out_channels,
            "kernel_size": kernel_size,
            "stride": stride,
            "padding": padding,
            "dilation": dilation,
            "groups": groups,
            "bias": bias,
        }
        super().__init__(info)
        self._unify()
        self.allow_frac = allow_frac

        if not self.allow_frac:
            in_channels = self.in_channels
            assert in_channels % self.groups == 0, (in_channels, self.groups)

    def get_output_shape(self, input_shape):
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -