code: def get_model_input_dtype(precision):
      if precision in ("16-mixed", "16", 16):
          return torch.float16
      if precision in ("bf16-mixed", "bf16"):
          return torch.bfloat16
      if precision in ("64-true", "64", 64):
          return torch.double
      return torch.float32
  
file_path: pytorch-lightning/tests/parity_fabric/utils.py
imports: import os
  import torch
  from lightning.fabric.accelerators.cuda import _clear_cuda_memory
label_desc: Use image node_id 5 for calling a global function with example usage: get_model_input_dtype(precision) and returns: torch, torch, torch, torch