tf.tpu.experimental.embedding.SparseCoreEmbeddingConfig

Config for sparsecore embedding.

View aliases

Compat aliases for migration

See Migration guide for more details.

tf.compat.v1.tpu.experimental.embedding.SparseCoreEmbeddingConfig

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,
    initialize_tables_on_host: bool = True,
    enable_fast_table_initialization: bool = False
)

Attributes
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
initialize_tables_on_host	Dataclass field
enable_fast_table_initialization	Dataclass field

Methods

__eq__

__eq__(
    other
)

Return self==value.

Class Variables
allow_id_dropping	False
disable_table_stacking	False
enable_fast_table_initialization	False
initialize_tables_on_host	True
max_ids_per_chip_per_sample	64
max_ids_per_table	None
max_unique_ids_per_table	None