Module: tf.data.experimental.service  |  TensorFlow v2.16.1 (original) (raw)

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.

Except as otherwise noted, the content of this page is licensed under the Creative Commons Attribution 4.0 License, and code samples are licensed under the Apache 2.0 License. For details, see the Google Developers Site Policies. Java is a registered trademark of Oracle and/or its affiliates. Some content is licensed under the numpy license.

Last updated 2024-04-26 UTC.