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