Docs4dev
Docs
tensorflow / 2.9 / raw_ops / dataservicedatasetv2.html

tf.raw_ops.DataServiceDatasetV2

Creates a dataset that reads data from the tf.data service.

tf.raw_ops.DataServiceDatasetV2(
    dataset_id,
    processing_mode,
    address,
    protocol,
    job_name,
    consumer_index,
    num_consumers,
    max_outstanding_requests,
    iteration_counter,
    output_types,
    output_shapes,
    task_refresh_interval_hint_ms=-1,
    data_transfer_protocol='',
    target_workers='AUTO',
    name=None
)
Args
dataset_id A Tensor of type int64.
processing_mode A Tensor of type string.
address A Tensor of type string.
protocol A Tensor of type string.
job_name A Tensor of type string.
consumer_index A Tensor of type int64.
num_consumers A Tensor of type int64.
max_outstanding_requests A Tensor of type int64.
iteration_counter A Tensor of type resource.
output_types A list of tf.DTypes that has length >= 1.
output_shapes A list of shapes (each a tf.TensorShape or list of ints) that has length >= 1.
task_refresh_interval_hint_ms An optional int. Defaults to -1.
data_transfer_protocol An optional string. Defaults to "".
target_workers An optional string. Defaults to "AUTO".
name A name for the operation (optional).
Returns
A Tensor of type variant.

© 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/DataServiceDatasetV2