code: def test_keras_wrong_placement_gpu():
      with tf.device("cpu:0"):
          model = mnist.keras_model()
          train_dataset = mnist.get_dataset("gpu", 0)
  
          model.fit(
              train_dataset,
              epochs=mnist.EPOCHS,
              steps_per_epoch=mnist.ITERATIONS,
          )
  
file_path: DALI/dali/test/python/test_dali_tf_dataset_mnist_graph.py
imports: import tensorflow
  import tensorflow.compat.v1
  from nose import with_setup
  import test_dali_tf_dataset_mnist
  from nose_utils import raises
label_desc: Use image node_id 4 for calling a global function with example usage: test_keras_wrong_placement_gpu() without return types