lingvo.core package
===================

.. automodule:: lingvo.core
   :members:
   :undoc-members:
   :show-inheritance:
   :private-members:

Subpackages
-----------

.. toctree::
   :maxdepth: 4

   lingvo.core.ops
   lingvo.core.steps

Submodules
----------

.. toctree::
   :maxdepth: 4

   lingvo.core.activations
   lingvo.core.adagraft
   lingvo.core.attention
   lingvo.core.attention_util
   lingvo.core.base_decoder
   lingvo.core.base_input_generator
   lingvo.core.base_layer
   lingvo.core.base_model
   lingvo.core.base_model_params
   lingvo.core.batch_major_attention
   lingvo.core.batch_utils
   lingvo.core.beam_search_helper
   lingvo.core.bfloat16_variables
   lingvo.core.bn_layers
   lingvo.core.build_data
   lingvo.core.builder
   lingvo.core.builder_layers
   lingvo.core.checkpointer
   lingvo.core.cluster
   lingvo.core.cluster_factory
   lingvo.core.computation_cost
   lingvo.core.conformer_layer
   lingvo.core.constants
   lingvo.core.conv_layers_builder
   lingvo.core.conv_layers_with_time_padding
   lingvo.core.datasource
   lingvo.core.datasource_tfds
   lingvo.core.decoder_lib
   lingvo.core.differentiable_assignment
   lingvo.core.distillation_task
   lingvo.core.distributed_shampoo
   lingvo.core.early_stop
   lingvo.core.egdd
   lingvo.core.entmax
   lingvo.core.favor_attention
   lingvo.core.flat_beam_search_helper
   lingvo.core.generic_input
   lingvo.core.gpipe
   lingvo.core.graddrop
   lingvo.core.gradient_combiner
   lingvo.core.gshard_builder
   lingvo.core.gshard_decode
   lingvo.core.gshard_layers
   lingvo.core.gshard_utils
   lingvo.core.hyperparams
   lingvo.core.hyperparams_pb2
   lingvo.core.inference_graph_exporter
   lingvo.core.inference_graph_pb2
   lingvo.core.input_generator_helper
   lingvo.core.input_policy
   lingvo.core.insertion
   lingvo.core.inspect_utils
   lingvo.core.layers
   lingvo.core.layers_with_attention
   lingvo.core.layers_with_gpipe
   lingvo.core.learner
   lingvo.core.lstm_frnn_layer
   lingvo.core.matrix_functions
   lingvo.core.metrics
   lingvo.core.ml_perf_bleu_metric
   lingvo.core.ml_perf_log
   lingvo.core.ml_perf_tokenizer
   lingvo.core.model_helper
   lingvo.core.multitask_model
   lingvo.core.nested_map
   lingvo.core.optimizer
   lingvo.core.plot
   lingvo.core.predictor
   lingvo.core.predictor_runner_base
   lingvo.core.program
   lingvo.core.program_utils
   lingvo.core.pruning_utils
   lingvo.core.py_utils
   lingvo.core.py_utils_flags
   lingvo.core.quant_test_lib
   lingvo.core.quant_utils
   lingvo.core.recurrent
   lingvo.core.repeat_layer
   lingvo.core.retry
   lingvo.core.reversible_layers
   lingvo.core.rnn_cell
   lingvo.core.rnn_layers
   lingvo.core.routing_config_helper
   lingvo.core.saver
   lingvo.core.scatter_update
   lingvo.core.schedule
   lingvo.core.scorers
   lingvo.core.self_attention_layer
   lingvo.core.sendrecv
   lingvo.core.spectrum_augmenter
   lingvo.core.spectrum_augmenter_on_device
   lingvo.core.step
   lingvo.core.stream_step_test_base
   lingvo.core.summary_utils
   lingvo.core.symbolic
   lingvo.core.target_sequence_sampler
   lingvo.core.task_scheduler
   lingvo.core.test_helper
   lingvo.core.test_trainer_utils
   lingvo.core.test_utils
   lingvo.core.thread_local_utils
   lingvo.core.tokenizers
   lingvo.core.tpu_embedding_layers
   lingvo.core.tpu_summary
   lingvo.core.trainer_test_utils
   lingvo.core.tshape
   lingvo.core.var_tmp_wrappers
   lingvo.core.wpm_encoder