parameters: (device)
return_statement: return torch_xla._XLAC._xla_memory_info(str(device))
docstring: Retrieves the device memory information.
  
    Args:
      device (string): The device whose memory information are requested.
  
    Returns:
      A dictionary with `kb_free` (free memory in KB) and `kb_total` (total
      memory in KB) keys.
docstring_summary: Retrieves the device memory information.
docstring_tokens: Retrieves the device memory information .
function: def get_memory_info(device):
    """Retrieves the device memory information.
  
    Args:
      device (string): The device whose memory information are requested.
  
    Returns:
      A dictionary with `kb_free` (free memory in KB) and `kb_total` (total
      memory in KB) keys.
    """
    return torch_xla._XLAC._xla_memory_info(str(device))
function_tokens: def get_memory_info ( device ) : return torch_xla . _XLAC . _xla_memory_info ( str ( device ) )