Index

_

__init__() (recis.features.feature.Feature method) (recis.features.FeatureEngine method) (recis.features.op.Bucketize method) (recis.features.op.FeatureCross method) (recis.features.op.Hash method) (recis.features.op.IDMultiHash method) (recis.features.op.Mod method) (recis.features.op.SelectField method) (recis.features.op.SelectFields method) (recis.features.op.SequenceTruncate method) (recis.framework.checkpoint_manager.CheckpointManager method) (recis.framework.checkpoint_manager.Saver method) (recis.framework.trainer.Trainer method) (recis.hooks.filter_hook.HashTableFilterHook method) (recis.hooks.profiler_hook.ProfilerHook method) (recis.metrics.auroc.AUROC method) (recis.metrics.gauc.Gauc method) (recis.nn.DynamicEmbedding method) (recis.nn.EmbeddingEngine method) (recis.nn.modules.hashtable.HashTable method) (recis.optim.adamw_tf.AdamWTF method) (recis.optim.sparse_adam.SparseAdam method) (recis.optim.sparse_adamw.SparseAdamW method) (recis.optim.sparse_adamw_tf.SparseAdamWTF method) (recis.serialize.checkpoint_reader.CheckpointReader method) (recis.serialize.loader.Loader method) (recis.serialize.saver.Saver method) (recis.utils.logger.Logger method)

_aucs (recis.metrics.gauc.Gauc attribute) _counts (recis.metrics.gauc.Gauc attribute) _cpu (recis.metrics.gauc.Gauc attribute) _word_size (recis.metrics.gauc.Gauc attribute)

A

accelerator (recis.framework.trainer.Trainer attribute) accept_grad() (recis.nn.modules.hashtable.HashTable method) AdamWTF (class in recis.optim.adamw_tf) add_hook() (recis.framework.trainer.Trainer method) add_hooks() (recis.framework.trainer.Trainer method) add_metric() (in module recis.framework.metrics)

add_op() (recis.features.feature.Feature method) admit_hook (recis.nn.EmbeddingOption attribute) AdmitHook (class in recis.nn.hashtable_hook) after_step() (recis.hooks.hook.Hook method) after_train() (recis.hooks.hook.Hook method) args (recis.framework.trainer.Trainer attribute) AUROC (class in recis.metrics.auroc)

B

before_step() (recis.hooks.hook.Hook method) before_train() (recis.hooks.hook.Hook method) block_size (recis.nn.EmbeddingOption attribute)

boundary (recis.features.op.Bucketize attribute) Bucketize (class in recis.features.op) bucketize() (in module recis.nn.functional.array_ops) bucketize_mod() (in module recis.nn.functional.array_ops)

C

checkpoint_manager (recis.framework.trainer.Trainer attribute) CheckpointManager (class in recis.framework.checkpoint_manager) CheckpointReader (class in recis.serialize.checkpoint_reader) children (recis.nn.EmbeddingOption attribute) children_info() (recis.nn.modules.hashtable.HashTable method) clear_grad() (recis.nn.modules.hashtable.HashTable method)

coalesced (recis.nn.EmbeddingOption attribute) combiner (recis.nn.EmbeddingOption attribute) combiner_kwargs (recis.nn.EmbeddingOption attribute) compute() (recis.metrics.auroc.AUROC method) ConstantInitializer (class in recis.nn.initializers) copy_data_to_device() (in module recis.utils.data_utils)

D

data_to_cuda (recis.framework.trainer.Trainer attribute) dense_lr_scheduler (recis.framework.trainer.Trainer attribute) dense_optimizer (recis.framework.trainer.Trainer attribute)

device (recis.nn.EmbeddingOption attribute) dist_sync_on_step (recis.metrics.auroc.AUROC attribute) dtype (recis.nn.EmbeddingOption attribute) DynamicEmbedding (class in recis.nn)

E

embedding_dim (recis.nn.EmbeddingOption attribute) EmbeddingEngine (class in recis.nn) EmbeddingOption (class in recis.nn) embeddings() (recis.nn.modules.hashtable.HashTable method)

error() (recis.utils.logger.Logger method) eval_dataset (recis.framework.trainer.Trainer attribute) eval_steps (recis.framework.trainer.TrainingArguments attribute) evaluate() (recis.framework.trainer.Trainer method)

F

farmhash() (in module recis.nn.functional.hash_ops) Feature (class in recis.features.feature) FeatureCross (class in recis.features.op) FeatureEngine (class in recis.features) filter_hook (recis.nn.EmbeddingOption attribute) filter_out_sparse_param() (in module recis.nn.modules.hashtable) FilterHook (class in recis.nn.hashtable_hook) fn (recis.metrics.auroc.AUROC attribute) forward() (recis.features.feature.Feature method) (recis.features.FeatureEngine method) (recis.features.op.FeatureCross method) (recis.features.op.Hash method) (recis.features.op.IDMultiHash method) (recis.features.op.Mod method) (recis.features.op.SelectField method) (recis.features.op.SelectFields method) (recis.features.op.SequenceTruncate method) (recis.metrics.auroc.AUROC method) (recis.metrics.gauc.Gauc method) (recis.nn.DynamicEmbedding method) (recis.nn.EmbeddingEngine method) (recis.nn.modules.hashtable.HashTable method)

fp (recis.metrics.auroc.AUROC attribute) fused_bucketize_gpu() (in module recis.nn.functional.fused_ops) fused_ids_encode_gpu() (in module recis.nn.functional.fused_ops) fused_multi_hash() (in module recis.nn.functional.fused_ops), [1] fused_ragged_cutoff_2D() (in module recis.nn.functional.ragged_ops) fused_ragged_cutoff_3D() (in module recis.nn.functional.ragged_ops) fused_uint64_mod_gpu() (in module recis.nn.functional.fused_ops)

G

Gauc (class in recis.metrics.gauc) gen_slice() (in module recis.nn.modules.hashtable)

grad() (recis.nn.modules.hashtable.HashTable method) grad_reduce_by (recis.nn.EmbeddingOption attribute) gradient_accumulation_steps (recis.framework.trainer.TrainingArguments attribute)

H

Hash (class in recis.features.op) hash_buckets (recis.io.OrcDataset attribute) hash_features (recis.io.OrcDataset attribute) hash_type (recis.features.op.Hash attribute)

hash_types (recis.io.OrcDataset attribute) HashTable (class in recis.nn.modules.hashtable) HashTableFilterHook (class in recis.hooks.filter_hook) Hook (class in recis.hooks.hook) hooks (recis.framework.trainer.Trainer attribute)

I

IDMultiHash (class in recis.features.op) ids() (recis.nn.modules.hashtable.HashTable method) ids_encode() (in module recis.nn.functional.embedding_ops) ids_partition() (in module recis.nn.functional.embedding_ops)

info() (recis.utils.logger.Logger method) initializer (recis.nn.EmbeddingOption attribute) input_list (recis.features.op.SelectFields attribute) insert() (recis.nn.modules.hashtable.HashTable method)

L

load() (recis.serialize.loader.Loader method) Loader (class in recis.serialize.loader) log_steps (recis.framework.trainer.TrainingArguments attribute)

Logger (class in recis.utils.logger) logger (recis.hooks.profiler_hook.ProfilerHook attribute) LoggerHook (class in recis.hooks.logger_hook)

M

max_partition_num (recis.nn.EmbeddingOption attribute) max_to_keep (recis.framework.trainer.TrainingArguments attribute) merge_offsets() (in module recis.nn.functional.embedding_ops) Mod (class in recis.features.op) mod (recis.features.op.Mod attribute) model (recis.framework.trainer.Trainer attribute) model_bank (recis.framework.trainer.TrainingArguments attribute)

module recis.nn.functional.embedding_ops recis.nn.functional.fused_ops recis.nn.functional.hash_ops recis.nn.functional.ragged_ops multi_hash() (in module recis.nn.functional.array_ops) murmurhash() (in module recis.nn.functional.hash_ops)

N

NormalInitializer (class in recis.nn.initializers)

num_thresholds (recis.metrics.auroc.AUROC attribute)

O

OrcDataset (class in recis.io)

output_dir (recis.framework.trainer.TrainingArguments attribute) (recis.hooks.profiler_hook.ProfilerHook attribute)

P

pg (recis.nn.EmbeddingOption attribute)

prof (recis.hooks.profiler_hook.ProfilerHook attribute) ProfilerHook (class in recis.hooks.profiler_hook)

R

ragged_embedding_segment_reduce() (in module recis.nn.functional.embedding_ops) read_tensor() (recis.serialize.checkpoint_reader.CheckpointReader method) reader (recis.serialize.checkpoint_reader.CheckpointReader attribute) recis.nn.functional.embedding_ops module recis.nn.functional.fused_ops module

recis.nn.functional.hash_ops module recis.nn.functional.ragged_ops module register_for_checkpointing() (recis.framework.checkpoint_manager.Saver method) reset() (recis.metrics.auroc.AUROC method) (recis.metrics.gauc.Gauc method) restore() (recis.framework.trainer.Trainer method)

S

save() (recis.framework.checkpoint_manager.CheckpointManager method) (recis.framework.checkpoint_manager.Saver method) (recis.serialize.saver.Saver method) save_concurrency_per_rank (recis.framework.trainer.TrainingArguments attribute) save_steps (recis.framework.trainer.TrainingArguments attribute) Saver (class in recis.framework.checkpoint_manager) (class in recis.serialize.saver) SelectField (class in recis.features.op) SelectFields (class in recis.features.op) SequenceTruncate (class in recis.features.op) set_grad_accum_steps() (recis.optim.sparse_adam.SparseAdam method) (recis.optim.sparse_adamw.SparseAdamW method) (recis.optim.sparse_adamw_tf.SparseAdamWTF method)

shared_name (recis.nn.EmbeddingOption attribute) slot_group() (recis.nn.modules.hashtable.HashTable method) sparse_embedding_segment_reduce() (in module recis.nn.functional.embedding_ops) sparse_optimizer (recis.framework.trainer.Trainer attribute) SparseAdam (class in recis.optim.sparse_adam) SparseAdamW (class in recis.optim.sparse_adamw) SparseAdamWTF (class in recis.optim.sparse_adamw_tf) step() (recis.optim.adamw_tf.AdamWTF method) (recis.optim.sparse_adam.SparseAdam method) (recis.optim.sparse_adamw.SparseAdamW method) (recis.optim.sparse_adamw_tf.SparseAdamWTF method)

T

tensor_dtype() (recis.serialize.checkpoint_reader.CheckpointReader method) tensor_names() (recis.serialize.checkpoint_reader.CheckpointReader method) tensor_shape() (recis.serialize.checkpoint_reader.CheckpointReader method) thresholds (recis.metrics.auroc.AUROC attribute) tn (recis.metrics.auroc.AUROC attribute) tp (recis.metrics.auroc.AUROC attribute) train() (recis.framework.trainer.Trainer method)

train_dataset (recis.framework.trainer.Trainer attribute) train_epoch (recis.framework.trainer.TrainingArguments attribute) train_steps (recis.framework.trainer.TrainingArguments attribute) trainable (recis.nn.EmbeddingOption attribute) Trainer (class in recis.framework.trainer) TrainingArguments (class in recis.framework.trainer) TruncNormalInitializer (class in recis.nn.initializers)

U

UniformInitializer (class in recis.nn.initializers)

update() (recis.metrics.auroc.AUROC method) use_weight (recis.nn.EmbeddingOption attribute)

W

warning() (recis.utils.logger.Logger method)

Z

zero_grad() (recis.optim.adamw_tf.AdamWTF method) (recis.optim.sparse_adam.SparseAdam method) (recis.optim.sparse_adamw.SparseAdamW method) (recis.optim.sparse_adamw_tf.SparseAdamWTF method)