lingvo package

Subpackages

• lingvo.core package
  • Subpackages
    • lingvo.core.ops package
      • Submodules
    • lingvo.core.steps package
      • Submodules
  • Submodules
    • lingvo.core.activations module
    • lingvo.core.adagraft module
      • AdaGraftOptimizer
    • lingvo.core.attention module
    • lingvo.core.attention_util module
    • lingvo.core.base_decoder module
    • lingvo.core.base_input_generator module
    • lingvo.core.base_layer module
    • lingvo.core.base_model module
    • lingvo.core.base_model_params module
    • lingvo.core.batch_major_attention module
    • lingvo.core.batch_utils module
    • lingvo.core.beam_search_helper module
    • lingvo.core.bfloat16_variables module
      • Bfloat16VariableSaveable
      • get_saver_spec_for_variables_with_bf16_overrides()
    • lingvo.core.bn_layers module
    • lingvo.core.build_data module
      • BuildData()
    • lingvo.core.builder module
    • lingvo.core.builder_layers module
    • lingvo.core.checkpointer module
    • lingvo.core.cluster module
      • InfeedContext
      • InfeedContextScope()
      • GetInfeedContext()
      • MakeDeviceString()
      • _Cluster
      • VarPlacer
      • _LeastLoadedPlacer
      • ParseDeviceString()
    • lingvo.core.cluster_factory module
      • SetCluster()
      • Current()
      • ForTestingWorker()
      • SetEval()
      • SetRequireSequentialInputOrder()
      • SetModelSplit()
    • lingvo.core.computation_cost module
    • lingvo.core.conformer_layer module
    • lingvo.core.constants module
    • lingvo.core.conv_layers_builder module
    • lingvo.core.conv_layers_with_time_padding module
    • lingvo.core.datasource module
    • lingvo.core.datasource_tfds module
    • lingvo.core.decoder_lib module
    • lingvo.core.differentiable_assignment module
      • max_assignment()
    • lingvo.core.distillation_task module
    • lingvo.core.distributed_shampoo module
    • lingvo.core.early_stop module
    • lingvo.core.egdd module
      • EGDD
    • lingvo.core.entmax module
      • _calculate_probability()
      • entmax_support()
      • entmax_loss()
    • lingvo.core.favor_attention module
    • lingvo.core.flat_beam_search_helper module
    • lingvo.core.generic_input module
    • lingvo.core.gpipe module
    • lingvo.core.graddrop module
    • lingvo.core.gradient_combiner module
    • lingvo.core.gshard_builder module
    • lingvo.core.gshard_decode module
    • lingvo.core.gshard_layers module
    • lingvo.core.gshard_utils module
      • Split()
      • Replicate()
      • GetMeshSplitSharding()
      • MeshSplit()
      • MeshSplitDimPrefixContext()
      • GetMeshSplitDimPrefixContext()
      • ManualMeshDimContext()
      • ZigzagOrderOnDeviceMesh()
      • GetNonPod2dMesh()
      • ReshapeDim()
      • TensorShardingSpec
      • GetVarSharding()
      • LoadSpm()
    • lingvo.core.hyperparams module
      • _QuoteString()
      • _UnquoteString()
      • _EndsWithTerminalQuote()
      • _IsNamedTuple()
      • _SortedDict
      • _Param
      • CopyFieldsTo()
      • Params
      • InstantiableParams
    • lingvo.core.hyperparams_pb2 module
      • _b()
      • Hyperparam
      • HyperparamRepeated
      • HyperparamValue
      • ProtoVal
      • EnumVal
      • NamedTupleVal
    • lingvo.core.inference_graph_exporter module
    • lingvo.core.inference_graph_pb2 module
      • _b()
      • InferenceGraph
    • lingvo.core.input_generator_helper module
    • lingvo.core.input_policy module
      • Apply()
    • lingvo.core.insertion module
    • lingvo.core.inspect_utils module
      • _IsDefinableParameter()
      • _ExtractParameters()
      • DefineParams()
      • _MakeArgs()
      • CallWithParams()
      • ConstructWithParams()
    • lingvo.core.layers module
    • lingvo.core.layers_with_attention module
    • lingvo.core.layers_with_gpipe module
    • lingvo.core.learner module
    • lingvo.core.lstm_frnn_layer module
    • lingvo.core.matrix_functions module
      • matrix_square_root()
      • inlined_matrix_inverse_pth_root()
    • lingvo.core.metrics module
    • lingvo.core.ml_perf_bleu_metric module
    • lingvo.core.ml_perf_log module
      • get_caller()
      • mlperf_format()
      • mlperf_print()
    • lingvo.core.ml_perf_tokenizer module
    • lingvo.core.model_helper module
    • lingvo.core.multitask_model module
    • lingvo.core.nested_map module
      • _FromNestedDict()
      • NestedMap
    • lingvo.core.optimizer module
    • lingvo.core.plot module
    • lingvo.core.predictor module
    • lingvo.core.predictor_runner_base module
    • lingvo.core.program module
    • lingvo.core.program_utils module
      • SummaryToCsv()
      • CsvToSummary()
      • DecodeStatusCache
      • TriggerScheduler
    • lingvo.core.pruning_utils module
    • lingvo.core.py_utils module
    • lingvo.core.py_utils_flags module
      • _FromGlobal()
      • enable_asserts()
      • enable_check_numerics()
      • use_xla()
      • use_tpu()
      • testonly_skip_norm_layers()
      • tpu_compat()
      • use_stateless_vars_init()
    • lingvo.core.quant_test_lib module
    • lingvo.core.quant_utils module
    • lingvo.core.recurrent module
    • lingvo.core.repeat_layer module
    • lingvo.core.retry module
      • Retry()
    • lingvo.core.reversible_layers module
    • lingvo.core.rnn_cell module
    • lingvo.core.rnn_layers module
    • lingvo.core.routing_config_helper module
    • lingvo.core.saver module
    • lingvo.core.scatter_update module
    • lingvo.core.schedule module
    • lingvo.core.scorers module
      • _ToUnicode()
      • _Tokenize()
      • NGrams()
      • Unsegmenter
      • BleuScorer
    • lingvo.core.self_attention_layer module
    • lingvo.core.sendrecv module
      • _TpuCore()
      • Channel
    • lingvo.core.spectrum_augmenter module
    • lingvo.core.spectrum_augmenter_on_device module
    • lingvo.core.step module
    • lingvo.core.stream_step_test_base module
    • lingvo.core.summary_utils module
    • lingvo.core.symbolic module
      • Symbol
      • IsSymbol()
      • IsExpr()
      • _LocalSymbolToValueStack
      • SymbolToValueMap
      • EvalExpr()
      • ToStatic()
      • ToTensor()
    • lingvo.core.target_sequence_sampler module
    • lingvo.core.task_scheduler module
    • lingvo.core.test_helper module
      • test_src_dir_path()
    • lingvo.core.test_trainer_utils module
    • lingvo.core.test_utils module
    • lingvo.core.thread_local_utils module
      • ThreadLocalStack
      • ThreadLocalDict
    • lingvo.core.tokenizers module
    • lingvo.core.tpu_embedding_layers module
    • lingvo.core.tpu_summary module
    • lingvo.core.trainer_test_utils module
    • lingvo.core.tshape module
      • Shape
    • lingvo.core.var_tmp_wrappers module
      • VarWrapperTrackAssign
      • StackedVarWrapperWithManualSharding
    • lingvo.core.wpm_encoder module
• lingvo.tasks package
  • Subpackages
    • lingvo.tasks.asr package
      • Subpackages
      • Submodules
    • lingvo.tasks.image package
      • Subpackages
      • Submodules
    • lingvo.tasks.lm package
      • Subpackages
      • Submodules
    • lingvo.tasks.milan namespace
      • Subpackages
      • Submodules
    • lingvo.tasks.mt package
      • Subpackages
      • Submodules
    • lingvo.tasks.punctuator package
      • Subpackages
      • Submodules
• lingvo.tools package
  • Submodules
    • lingvo.tools.audio_lib module
    • lingvo.tools.beam_utils module
      • BeamInit()
      • GetPipelineRoot()
      • GetReader()
      • GetWriter()
      • GetEmitterFn()
    • lingvo.tools.bpe_word_tokenizer module
      • _GetVocabulary()
      • _ExtractTokenization()
      • main()
    • lingvo.tools.compare_params module
    • lingvo.tools.compute_stats module
      • StatsCollector
      • main()
    • lingvo.tools.count_records module
      • main()
    • lingvo.tools.create_asr_features module
    • lingvo.tools.gke_launch module
      • _get_or_add()
      • add_gpu_to_pod()
      • set_pod_cpu_memory()
      • decoder_template()
      • _tpu_resource()
      • tpu_training_template()
      • tensorboard_template()
      • build_docker_image()
      • get_gke_cluster()
      • create_tpu_cluster()
      • validate_args()
      • main()
    • lingvo.tools.keras2ckpt module
      • main()
    • lingvo.tools.print_tf_records module
      • _ListDebugString()
      • _CustomShortDebugString()
      • _PrintHeader()
      • _PrintFiles()
      • main()
    • lingvo.tools.wpm_encode_file module

Submodules

• lingvo.base_runner module
• lingvo.base_trial module
  • Trial
    • Trial.Params()
    • Trial.report_interval_seconds
    • Trial.objective_metric_key
    • Trial.Name()
    • Trial.OverrideModelParams()
    • Trial.ShouldStop()
    • Trial.ReportDone()
    • Trial.ShouldStopAndMaybeReport()
    • Trial._DoReportTrainingProgress()
    • Trial.ReportEvalMeasure()
  • NoOpTrial
    • NoOpTrial.Name()
    • NoOpTrial.OverrideModelParams()
    • NoOpTrial.ShouldStop()
    • NoOpTrial.ReportDone()
    • NoOpTrial.ShouldStopAndMaybeReport()
    • NoOpTrial.ReportEvalMeasure()
  • TunerManagedError
• lingvo.compat module
  • _clone_module()
  • summarize_tf2_status()
  • _CacheDataset
    • _CacheDataset._abc_impl
  • _ShuffleDataset
    • _ShuffleDataset._abc_impl
  • stateless_shuffle_dataset()
  • stateless_cache_dataset()
  • stateless_list_files()
  • variable_scope
• lingvo.datasets module
  • DatasetFunctionError
  • GetAllDatasetParamsNotImplementedError
  • GetDatasets()
  • GetDatasetsAst()
• lingvo.eager_runners module
• lingvo.executor module
• lingvo.ipython_kernel module
• lingvo.model_imports module
  • _Import()
  • _InsertParams()
  • ImportAllParams()
  • ImportParams()
• lingvo.model_registry module
• lingvo.models_test_helper module
• lingvo.pdb_wrapper module
  • catch_post_mortem()
• lingvo.runners module
• lingvo.trainer module
• lingvo.trainer_utils module