message: add select_device argument
diff: @@ -33,7 +33,7 @@ FLAGS = None
  def main(_):
  - with tf.device('/device:NGRAPH:0'):
  + with tf.device('/device:'+FLAGS.select_device+':0'):
  run_mnist(_)
  @@ -106,5 +106,11 @@ if __name__ == '__main__':
  default=1000,
  help='Number of training iterations')
  + parser.add_argument(
  + '--select_device',
  + type=str,
  + default='NGRAPH',
  + help='Enter the device')
  +
  FLAGS, unparsed = parser.parse_known_args()
  tf.app.run(main=main, argv=[sys.argv[0]] + unparsed)
  