W3cubDocs

/TensorFlow 2.4

Module: tf.distribute.experimental.partitioners

Public API for tf.distribute.experimental.partitioners namespace.

Classes

class FixedShardsPartitioner: Partitioner that allocates a fixed number of shards.

class MaxSizePartitioner: Partitioner that keeps shards below max_shard_bytes.

class MinSizePartitioner: Partitioner that allocates a minimum size per shard.

class Partitioner: Partitioner base class: all partitiners inherit from this class.

© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.4/api_docs/python/tf/distribute/experimental/partitioners