add nn_device function
nn_device