code: def tensor_round(tensor: Tensor) -> Tensor:
      if tensor.device.type == "cpu" and tensor.dtype == torch.float16:
          # CPU does not support FP16 round
          return (
              tensor.to(dtype=torch.float32)
              .round_()
              .to(dtype=torch.float16)
          )
      else:
          return tensor.round_()
  
imports: import torch
  import deepspeed
  from torch import Tensor
  from typing import Tuple
  import torch.nn
  from typing import Dict, Callable, Union
  from deepspeed.accelerator import get_accelerator
  import functools
label_desc: Use image node_id 3 for calling a global function with example usage: tensor_round(tensor) and returns: tensor, tensor