SparseOperationKit
1.1.0
  • Introduction
  • Features
  • Get Started
  • Examples
  • Performance
  • API Docs
  • Release Notes
  • Environment Variables
  • Known Issues
SparseOperationKit
  • »
  • Index

Index

A | C | D | I | L | M | O | R | S | T

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)

© Copyright 2022, NVIDIA.

Built with Sphinx using a theme provided by Read the Docs.
Other Versions master: sok_v1.1.0
Tags
sok_v1.0.0
sok_v1.1.0
sok_v1.1.1
sok_v1.1.2
sok_v1.1.4