def METHOD_NAME(test_case):
    # different mask shape for each mode
    _test_fused_scale_mask_bias_softmax(test_case, 16, 128, 64, 8, 32, "row")
    _test_fused_scale_mask_bias_softmax(test_case, 16, 128, 64, 8, 32, "col")
    _test_fused_scale_mask_bias_softmax(
        test_case, 16, 128, 64, 8, 32, "triangular_start"
    )
    _test_fused_scale_mask_bias_softmax(
        test_case, 16, 128, 64, 8, 32, "triangular_end"
    )
    _test_fused_scale_mask_bias_softmax(test_case, 16, 128, 64, 8, 32, "template")
    _test_fused_scale_mask_bias_softmax(test_case, 16, 128, 64, 8, 32, "global_col")
    _test_fused_scale_mask_bias_softmax(test_case, 16, 128, 64, 8, 32, "ensemble")
    _test_fused_scale_mask_bias_softmax(
        test_case, 16, 128, 64, 8, 32, "row", inplace=True
    )
    _test_fused_scale_mask_bias_softmax(
        test_case, 16, 128, 64, 8, 32, "col", inplace=True
    )
    _test_fused_scale_mask_bias_softmax(
        test_case, 128, 128, 64, 8, 32, "triangular_start", inplace=True
    )
    _test_fused_scale_mask_bias_softmax(
        test_case, 16, 128, 64, 8, 32, "triangular_end", inplace=True
    )
    _test_fused_scale_mask_bias_softmax(
        test_case, 16, 128, 64, 8, 32, "template", inplace=True
    )
    _test_fused_scale_mask_bias_softmax(
        test_case, 16, 128, 64, 8, 32, "global_col", inplace=True
    )
    _test_fused_scale_mask_bias_softmax(
        test_case, 16, 128, 64, 8, 32, "ensemble", inplace=True
    )