code: def _get_embed_device(vocab_size):
      """Decide on which device to place an embed matrix given its vocab size."""
      if vocab_size > VOCAB_SIZE_THRESHOLD_CPU:
          return "/cpu:0"
      else:
          return "/gpu:0"
  
imports: from __future__ import print_function
  import collections
  import os
  import time
  import numpy
  import six
  import tensorflow
  from tensorflow.python.ops import lookup_ops
  from .utils import misc_utils, vocab_utils, iterator_utils
label_desc: Use image node_id 6 for calling a global function with example usage: _get_embed_device(vocab_size) and returns: str, str