tf.raw_ops.ThreadPoolHandle
Creates a dataset that uses a custom thread pool to compute input_dataset.
tf.raw_ops.ThreadPoolHandle(
num_threads,
display_name,
max_intra_op_parallelism=1,
container='',
shared_name='',
name=None
)
| Args | |
|---|---|
num_threads |
An int. The number of threads in the thread pool. |
display_name |
A string. A human-readable name for the threads that may be visible in some visualizations. threadpool. |
max_intra_op_parallelism |
An optional int. Defaults to 1. The maximum degree of parallelism to use within operations that execute on this threadpool. |
container |
An optional string. Defaults to "". |
shared_name |
An optional string. Defaults to "". |
name |
A name for the operation (optional). |
| Returns | |
|---|---|
A Tensor of type resource. |
© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.9/api_docs/python/tf/raw_ops/ThreadPoolHandle