tf.tpu.experimental.embedding.SparseCoreEmbeddingConfig

Config for sparsecore embedding.

tf.tpu.experimental.embedding.SparseCoreEmbeddingConfig(
    disable_table_stacking: bool = False,
    max_ids_per_chip_per_sample: int = 64,
    max_ids_per_table: Optional[Dict[str, int]] = None,
    max_unique_ids_per_table: Optional[Dict[str, int]] = None,
    allow_id_dropping: bool = False
)

disable_table_stacking Dataclass field
max_ids_per_chip_per_sample Dataclass field
max_ids_per_table Dataclass field
max_unique_ids_per_table Dataclass field
allow_id_dropping Dataclass field

Methods

__eq__

__eq__(
    other
)

Return self==value.

allow_id_dropping False
disable_table_stacking False
max_ids_per_chip_per_sample 64
max_ids_per_table None
max_unique_ids_per_table None