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]
def device_of_devicendarray(devicendarray): dev = device_of_gpu_matrix(devicendarray) return get_visible_devices()[dev]