title: How to Remove additional information Executing op __inference_train_function_88100 in device in Tensorflow version >2.0
body: <p>I was just testing the code to verify whether the code is running in GPU or not and I got this additional information along with accuracy &amp; loss info.</p>
  <pre><code>Executing op __inference_train_function_88100 in device /job:localhost/replica:0/task:0/device:GPU:0
  Executing op __inference_train_function_88100 in device /job:localhost/replica:0/task:0/device:GPU:0
    81/1875 [&gt;.............................] - ETA: 9s - loss: 1.5444 - accuracy: 0.5918Executing op __inference_train_function_88100 in device /job:localhost/replica:0/task:0/device:GPU:0
  Executing op __inference_train_function_88100 in device /job:localhost/replica:0/task:0/device:GPU:0
  Executing op __inference_train_function_88100 in device /job:localhost/replica:0/task:0/device:GPU:0
  Executing op __inference_train_function_88100 in device /job:localhost/replica:0/task:0/device:GPU:0
  </code></pre>
  <p>The Code is given below (and code is running on jupyter notebook)</p>
  <pre><code>import os
  tf.autograph.set_verbosity(0)
  
  with tf.device(&quot;/gpu:0&quot;):
      model = keras.Sequential([
          keras.layers.Dense(10, input_shape=(784,), activation='sigmoid')
      ])
  
      model.compile(optimizer='adam',
                    loss='sparse_categorical_crossentropy',
                    metrics=['accuracy'])
  
      model.fit(X_train_flattened, y_train, epochs=5)
  </code></pre>
  <p>I also referred to TensorFlow documentation but, none of the methods worked.
  if you have any suggestions then, please help me</p>