Path	Lines of Code
lingvo/core/__init__.py	1
lingvo/core/activations.py	76
lingvo/core/adagraft.py	103
lingvo/core/attention.py	1945
lingvo/core/attention_util.py	450
lingvo/core/base_decoder.py	347
lingvo/core/base_input_generator.py	1109
lingvo/core/base_layer.py	568
lingvo/core/base_model.py	962
lingvo/core/base_model_params.py	62
lingvo/core/batch_major_attention.py	4756
lingvo/core/batch_utils.py	27
lingvo/core/beam_search_helper.py	503
lingvo/core/bfloat16_variables.py	31
lingvo/core/bn_layers.py	716
lingvo/core/build_data.py	4
lingvo/core/builder.py	168
lingvo/core/builder_layers.py	953
lingvo/core/checkpointer.py	441
lingvo/core/cluster.py	419
lingvo/core/cluster_factory.py	54
lingvo/core/computation_cost.py	34
lingvo/core/conformer_layer.py	871
lingvo/core/constants.py	1
lingvo/core/conv_layers_builder.py	408
lingvo/core/conv_layers_with_time_padding.py	540
lingvo/core/datasource.py	648
lingvo/core/decoder_lib.py	4
lingvo/core/distillation_task.py	121
lingvo/core/distributed_shampoo.py	456
lingvo/core/early_stop.py	138
lingvo/core/egdd.py	102
lingvo/core/entmax.py	70
lingvo/core/favor_attention.py	289
lingvo/core/flat_beam_search_helper.py	260
lingvo/core/generic_input.py	72
lingvo/core/gpipe.py	413
lingvo/core/graddrop.py	98
lingvo/core/gradient_combiner.py	4
lingvo/core/gshard_builder.py	2949
lingvo/core/gshard_decode.py	360
lingvo/core/gshard_layers.py	1697
lingvo/core/gshard_utils.py	285
lingvo/core/hyperparams.proto	56
lingvo/core/hyperparams.py	668
lingvo/core/inference_graph.proto	104
lingvo/core/inference_graph_exporter.py	332
lingvo/core/input_generator_helper.py	47
lingvo/core/input_policy.py	10
lingvo/core/insertion.py	156
lingvo/core/inspect_utils.py	41
lingvo/core/layers.py	4088
lingvo/core/layers_with_attention.py	1882
lingvo/core/layers_with_gpipe.py	1119
lingvo/core/learner.py	338
lingvo/core/lstm_frnn_layer.py	101
lingvo/core/matrix_functions.py	79
lingvo/core/metrics.py	349
lingvo/core/ml_perf_bleu_metric.py	128
lingvo/core/ml_perf_log.py	46
lingvo/core/ml_perf_tokenizer.py	12
lingvo/core/model_helper.py	32
lingvo/core/multitask_model.py	64
lingvo/core/nested_map.py	239
lingvo/core/optimizer.py	823
lingvo/core/plot.py	295
lingvo/core/predictor.py	234
lingvo/core/predictor_runner_base.py	183
lingvo/core/program.py	668
lingvo/core/pruning_utils.py	118
lingvo/core/py_utils.py	3563
lingvo/core/py_utils_flags.py	76
lingvo/core/quant_test_lib.py	92
lingvo/core/quant_utils.py	757
lingvo/core/recurrent.py	884
lingvo/core/repeat_layer.py	123
lingvo/core/retry.py	43
lingvo/core/reversible_layers.py	114
lingvo/core/rnn_cell.py	1839
lingvo/core/rnn_layers.py	892
lingvo/core/routing_config_helper.py	93
lingvo/core/saver.py	294
lingvo/core/scatter_update.py	31
lingvo/core/schedule.py	622
lingvo/core/scorers.py	69
lingvo/core/self_attention_layer.py	79
lingvo/core/sendrecv.py	64
lingvo/core/spectrum_augmenter.py	539
lingvo/core/spectrum_augmenter_on_device.py	25
lingvo/core/step.py	297
lingvo/core/stream_step_test_base.py	166
lingvo/core/summary_utils.py	179
lingvo/core/symbolic.py	53
lingvo/core/target_sequence_sampler.py	163
lingvo/core/task_scheduler.py	209
lingvo/core/thread_local_utils.py	9
lingvo/core/tokenizers.py	281
lingvo/core/tpu_embedding_layers.py	875
lingvo/core/tpu_summary.py	163
lingvo/core/trainer_test_utils.py	82
lingvo/core/tshape.py	53
lingvo/core/var_tmp_wrappers.py	106
lingvo/core/wpm_encoder.py	108