tf.train.experimental.ShardableTensor

Tensor wrapper containing data necessary for sharding.

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__