Public API for tf._api.v2.data.experimental.service namespace
Classes
class CrossTrainerCache
: Options related to the tf.data service cross trainer cache.
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.
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.