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. |
© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.4/api_docs/python/tf/raw_ops/ThreadPoolHandle