class_docstr: Policy that splits tensors into shards with a max shard size.
code: def description(self) -> str:
      return "Split tensors into shards with a 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 2 for calling the MaxShardSizePolicy obj's underlying member method code with example usage: obj.description() and returns: str