code: def get_one_device():
      return tf.distribute.OneDeviceStrategy(device="/gpu:0")
  
imports: import pickle
  import tensorflow
  from official.core import train_lib
label_desc: Use image node_id 3 for calling a global function with example usage: get_one_device() and returns: tf