I’d like to pass the TensorFlow intraop-threads and interop-threads arguments to the session configuration. It’d be great if there was a command line switch to do this. For example,
# Optimize CPU threads for TensorFlow
config = tf.ConfigProto(
inter_op_parallelism_threads=args.interop_threads,
intra_op_parallelism_threads=args.intraop_threads)
sess = tf.Session(config=config)