code: def __init__(
      self,
  ):
      super(triton_flash_attn, self).__init__()
  
file_path: deepspeed/deepspeed/ops/transformer/inference/triton_ops.py
imports: import torch
  import triton
  import triton.language
label_desc: Use image node_id 1 to create a new triton_flash_attn object from inherited base classes: torch.nn with example: obj = triton_flash_attn()