tf.train.experimental.ShardableTensor

Tensor wrapper containing data necessary for sharding.

tf.train.experimental.ShardableTensor(
    _tensor_save_spec: saveable_object.SaveSpec,
    tensor: tf.Tensor,
    dtype: tf.dtypes.DType,
    device: tf.DeviceSpec,
    name: str,
    shape: tf.TensorShape,
    slice_spec: tf.Variable.SaveSliceInfo,
    checkpoint_key: str,
    trackable: base.Trackable
)

The tensor representation used as inputs to pre-made and custom tf.train.experiemental.ShardingCallbacks, which can be specified using the experimental_sharding_callback option in tf.train.CheckpointOptions.

tensor Dataclass field
dtype Dataclass field
device Dataclass field
name Dataclass field
shape Dataclass field
slice_spec Dataclass field
checkpoint_key Dataclass field
trackable Dataclass field

Methods

__eq__

__eq__(
    other
)