tf.data.experimental.service.WorkerConfig

Configuration class for tf.data service disers.

tf.data.experimental.service.WorkerConfig(
    diser_address,
    worker_address=None,
    port=0,
    protocol=None,
    heartbeat_interval_ms=None,
    diser_timeout_ms=None,
    data_transfer_protocol=None,
    data_transfer_address=None
)

diser_addressSpecifies the address of the diser.
worker_addressSpecifies the address of the worker server. This address is passed to the diser so that the diser can tell clients how to connect to this worker.
portSpecifies the port to bind to. A value of 0 indicates that the worker can bind to any available port.
protocolA string indicating the protocol to be used by the worker to connect to the diser. E.g. "grpc".
heartbeat_interval_msHow often the worker should heartbeat to the diser, in milliseconds. If not set, the runtime will select a reasonable default. A higher value will reduce the load on the diser, while a lower value will reduce the time it takes to reclaim resources from finished jobs.
diser_timeout_msHow long, in milliseconds, to retry requests to the diser before giving up and reporting an error. Defaults to 1 hour.
data_transfer_protocolA string indicating the protocol to be used by the worker to transfer data to the client. E.g. "grpc".
data_transfer_addressA string indicating the data transfer address of the worker server.

diser_addressA namedtuple alias for field number 0
worker_addressA namedtuple alias for field number 1
portA namedtuple alias for field number 2
protocolA namedtuple alias for field number 3
heartbeat_interval_msA namedtuple alias for field number 4
diser_timeout_msA namedtuple alias for field number 5
data_transfer_protocolA namedtuple alias for field number 6
data_transfer_addressA namedtuple alias for field number 7