if bias_size is None and has_bias:
      bias_size = weight_shape[-1]

    # Verify that when bias_size is not None, has_bias should be True.
    # And if bias_size is None, has_bias should be False using XNOR
    assert not ((bias_size is not None) ^ has_bias)

    # Verify that bias size is correct
    if bias_size:
      input_height = input_shape[0] if len(input_shape) == 2 else input_shape[1]