Index

A

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

All2AllDenseEmbedding (class in sparse_operation_kit.embeddings.all2all_dense_embedding) assign() (in module sparse_operation_kit.experiment.dynamic_variable)

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) DynamicVariable (class in sparse_operation_kit.experiment.dynamic_variable)

E

export() (in module sparse_operation_kit.experiment.dynamic_variable)

I

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

init() (in module sparse_operation_kit.experiment)

L

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

lookup_sparse() (in module sparse_operation_kit.experiment.lookup)

M

module sparse_operation_kit.core.initialize sparse_operation_kit.experiment sparse_operation_kit.experiment.distributed_variable sparse_operation_kit.experiment.dynamic_variable sparse_operation_kit.experiment.lookup sparse_operation_kit.experiment.optimizer sparse_operation_kit.optimizers.utils

O

OptimizerScope (class in sparse_operation_kit.core.context_scope)

OptimizerWrapper() (in module sparse_operation_kit.experiment.optimizer)

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.experiment module sparse_operation_kit.experiment.distributed_variable module sparse_operation_kit.experiment.dynamic_variable module

sparse_operation_kit.experiment.lookup module sparse_operation_kit.experiment.optimizer 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)

V

Variable() (in module sparse_operation_kit.experiment.distributed_variable)