Exemple #1
0
def device_of_devicendarray(devicendarray):
    """
    Returns the device that backs memory allocated on the given
    deviceNDArray
    :param devicendarray: devicendarray array to check
    :return: int device id
    """
    dev = device_of_gpu_matrix(devicendarray)
    return get_visible_devices()[dev]
Exemple #2
0
def device_of_devicendarray(devicendarray):
    dev = device_of_gpu_matrix(devicendarray)
    return get_visible_devices()[dev]