if self._mode == 'fan_in':
            fan_in = f_in if self._fan_in is None else self._fan_in
        if self._mode == 'fan_out':
            fan_in = f_out

        if self._mode not in ['fan_in', 'fan_out']:
            raise ValueError(
                "The mode of KaimingNormal/KaimingUniform should be 'fan_in' or 'fan_out', "
                f"but received {self._mode}."
            )
        if self._mode == 'fan_out' and self._fan_in is not None:
            raise ValueError(
                "The mode of KaimingNormal/KaimingUniform is 'fan_out', "
                "but fan_in is set. Please set fan_in to None."
            )