tf.train.experimental.ShardableTensor

Tensor wrapper containing data necessary for sharding.

Compat aliases for migration

See Migration guide for more details.

tf.compat.v1.train.experimental.ShardableTensor

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__