tf.config.threading.set_intra_op_parallelism_threads

Set number of threads used within an individual op for parallelism.

View aliases

Compat aliases for migration

tf.compat.v1.config.threading.set_intra_op_parallelism_threads, `tf.compat.v2.config.threading.set_intra_op_parallelism_threads`

tf.config.threading.set_intra_op_parallelism_threads(
    num_threads
)

Certain operations like matrix multiplication and reductions can utilize parallel threads for speed ups. A value of 0 means the system picks an appropriate number.

Args
`num_threads`	Number of parallel threads

Except as otherwise noted, the content of this page is licensed under the Creative Commons Attribution 4.0 License, and code samples are licensed under the Apache 2.0 License. For details, see the Google Developers Site Policies. Java is a registered trademark of Oracle and/or its affiliates.

Last updated 2020-10-01 UTC.

tf.config.threading.set_intra_op_parallelism_threads

View aliases

Args