def METHOD_NAME(dtype):
  return {
  'int64': 'torch.int64',
  'int32': 'torch.int32',
  'float32': 'torch.float',
  'float64': 'torch.double',
  'torch.int64': 'int64',
  'torch.long': 'int64',
  'torch.int32': 'int32',
  'torch.int': 'int32',
  'torch.float32': 'float32',
  'torch.float': 'float32',
  'torch.float64': 'float64',
  'torch.double': 'float64',
  }.get(dtype, dtype)