def METHOD_NAME(
    input_a,
    input_b,
    kernel_size,
    max_displacement,
    stride_1,
    stride_2,
    pad,
    data_format,
):
    input_a_op = tf.convert_to_tensor(input_a, dtype=tf.float32)
    input_b_op = tf.convert_to_tensor(input_b, dtype=tf.float32)
    output = CorrelationCost(
        kernel_size=kernel_size,
        max_displacement=max_displacement,
        stride_1=stride_1,
        stride_2=stride_2,
        pad=pad,
        data_format=data_format,
    )([input_a_op, input_b_op])
    return output