Module: tf.data.experimental.service

Public API for tf._api.v2.data.experimental.service namespace

Classes

class CrossTrainerCache: Options related to the tf.data service cross trainer cache.

class DispatchServer: An in-process tf.data service dispatch server.

class DispatcherConfig: Configuration class for tf.data service dispatchers.

class ShardingPolicy: Specifies how to shard data among tf.data service workers.

class WorkerConfig: Configuration class for tf.data service dispatchers.

class WorkerServer: An in-process tf.data service worker server.

Functions

distribute(...): A transformation that moves dataset processing to the tf.data service.

from_dataset_id(...): Creates a dataset which reads data from the tf.data service.

register_dataset(...): Registers a dataset with the tf.data service.