code: def spec_augment_launch_heuristics(
      x: torch.Tensor, length: torch.Tensor
  ):
      """
      Heuristics to determins whether pytorch implementation or numba implementation is selected.
      Assumes numba cuda is supported.
  
      Args:
          x: Torch tensor of shape [B, F, T]
          length: Optional, Torch of tensor of shape [B] - containing lengths of the tensor.
  
      Returns:
          True if numba kernel should be selected, else False
      """
      if not x.is_cuda:
          return False
  
      if length is None:
          return False
  
      if x.shape[0] < 8:
          return False
  
      return True
  
file_path: NeMo/nemo/collections/asr/parts/numba/spec_augment/spec_aug_numba.py
imports: import torch
  import torch.nn
  from numba import cuda
  from nemo.core.classes import Typing, typecheck
  from nemo.core.neural_types import LengthsType, NeuralType, SpectrogramType
  from nemo.utils import logging
label_desc: Use image node_id 2 for calling a global function with example usage: spec_augment_launch_heuristics(x, length) and returns: True, False, False, False