Index

B

broadcast_variables() (in module distributed_embeddings.python.layers.dist_model_parallel)

build() (distributed_embeddings.python.layers.embedding.Embedding method)

C

compute_output_shape() (distributed_embeddings.python.layers.embedding.Embedding method)

D

distributed_embeddings.python module

DistributedEmbedding (class in distributed_embeddings.python.layers.dist_model_parallel) DistributedGradientTape() (in module distributed_embeddings.python.layers.dist_model_parallel)

E

Embedding (class in distributed_embeddings.python.layers.embedding)

embedding_lookup() (in module distributed_embeddings.python.ops.embedding_lookup_ops)

F

from_config() (distributed_embeddings.python.layers.embedding.Embedding class method)

G

get_config() (distributed_embeddings.python.layers.embedding.Embedding method)

get_weights() (distributed_embeddings.python.layers.dist_model_parallel.DistributedEmbedding method)

I

IntegerLookup (class in distributed_embeddings.python.layers.embedding)

M

module distributed_embeddings.python

S

set_weights() (distributed_embeddings.python.layers.dist_model_parallel.DistributedEmbedding method)