tf.config.threading.set_inter_op_parallelism_threads

View source on GitHub

Set number of threads used for parallelism between independent operations.

tf.config.threading.set_inter_op_parallelism_threads(
    num_threads
)

Determines the number of threads used by independent non-blocking operations. 0 means the system picks an appropriate number.

Args:

• num_threads: Number of parallel threads