Index

A

Adam (class in sparse_operation_kit.tf.keras.optimizers.adam)

All2AllDenseEmbedding (class in sparse_operation_kit.embeddings.all2all_dense_embedding)

C

call() (sparse_operation_kit.embeddings.all2all_dense_embedding.All2AllDenseEmbedding method) (sparse_operation_kit.embeddings.distributed_embedding.DistributedEmbedding method) (sparse_operation_kit.embeddings.tf_distributed_embedding.TFDistributedEmbedding method)

D

DistributedEmbedding (class in sparse_operation_kit.embeddings.distributed_embedding)

dump_to_file() (sparse_operation_kit.saver.Saver.Saver method)

I

Init() (in module sparse_operation_kit.core.initialize)

L

load_embedding_values() (sparse_operation_kit.saver.Saver.Saver method)

M

module sparse_operation_kit.core.initialize sparse_operation_kit.optimizers.utils

O

OptimizerScope (class in sparse_operation_kit.core.context_scope)

R

restore_from_file() (sparse_operation_kit.saver.Saver.Saver method)

S

Saver (class in sparse_operation_kit.saver.Saver) sparse_operation_kit.core.initialize module

sparse_operation_kit.optimizers.utils module split_embedding_variable_from_others() (in module sparse_operation_kit.optimizers.utils)

T

TFDistributedEmbedding (class in sparse_operation_kit.embeddings.tf_distributed_embedding)