def METHOD_NAME(tensor_parallel_degree, tensor_parallel_rank, hidden_size, intermediate_size):
    def fn(x):
        if x is None:
            return None
        x = merge_o_tensor_parallel_weight(
            x,
            tensor_parallel_degree=tensor_parallel_degree,
            hidden_size=hidden_size,
            intermediate_size=intermediate_size,
        )
        return x
    return fn