code: def _is_bias_type_supported_in_BMK(attn_bias_type: Any) -> bool:
      # NoneType
      if isinstance(None, attn_bias_type):
          return True
      if attn_bias_type in [LowerTriangularMask, torch.Tensor]:
          return True
      return False
  
imports: import math
  from dataclasses import dataclass
  from functools import partial
  from typing import Any, Callable, List, Mapping, Optional, Set, Tuple, Type, Union
  import torch
  from ..._cpp_lib import _built_with_cuda
  from ..common import BaseOperator
  from .attn_bias import AttentionBias, BlockDiagonalMask, LowerTriangularMask, LowerTriangularMaskWithTensorBias
label_desc: Use image node_id 1 for calling a global function with example usage: _is_bias_type_supported_in_BMK(attn_bias_type) and returns: False, True, True