def METHOD_NAME(
    self,
    data_type,
    input_shape,
    input_layout,
    output_shape,
    output_layout,
    hexagon_session,
):
    """Test batch flatten."""
    reshape_helper(
        "batch_flatten",
        sl.batch_flatten_compute,
        sl.batch_flatten_stir_schedule,
        data_type,
        input_shape,
        input_layout,
        output_shape,
        output_layout,
        hexagon_session,
    )