View source on GitHub |
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.ShardingCallback
s, which can be specified using the
experimental_sharding_callback
option in tf.train.CheckpointOptions
.
Methods
__eq__
__eq__(
other
)