class_docstr: Policy that splits tensors into shards with a max shard size.
code: def __init__(self, max_shard_size: int):
      self.max_shard_size = max_shard_size
  
imports: import math
  from typing import MutableSequence, Sequence
  from absl import logging
  from tensorflow.python.checkpoint.sharding import sharding_util
  from tensorflow.python.eager import context
  from tensorflow.python.framework import dtypes
  from tensorflow.python.framework import ops
  from tensorflow.python.framework import tensor
  from tensorflow.python.framework import tensor_shape
  from tensorflow.python.ops import array_ops
  from tensorflow.python.ops import string_ops
  from tensorflow.python.ops import variables
label_desc: Use image node_id 1 to create a new MaxShardSizePolicy object from inherited base classes: sharding_util with example: obj = MaxShardSizePolicy(max_shard_size)