Path	Lines of Code
lingvo/compat_test.py	32
lingvo/core/activations_test.py	42
lingvo/core/adagraft_test.py	42
lingvo/core/attention_test.py	2001
lingvo/core/attention_util_test.py	573
lingvo/core/base_decoder_test.py	115
lingvo/core/base_input_generator_test.py	464
lingvo/core/base_layer_test.py	289
lingvo/core/base_model_params_test.py	27
lingvo/core/base_model_test.py	466
lingvo/core/batch_major_attention_test.py	4237
lingvo/core/batch_utils_test.py	73
lingvo/core/beam_search_helper_test.py	398
lingvo/core/bfloat16_variables_test.py	30
lingvo/core/builder_layers_test.py	636
lingvo/core/builder_test.py	255
lingvo/core/checkpointer_eager_test.py	134
lingvo/core/checkpointer_test.py	186
lingvo/core/cluster_test.py	436
lingvo/core/conformer_layer_test.py	681
lingvo/core/conv_layers_builder_test.py	534
lingvo/core/conv_layers_with_time_padding_test.py	730
lingvo/core/datasource_test.py	645
lingvo/core/distributed_shampoo_test.py	270
lingvo/core/early_stop_test.py	153
lingvo/core/egdd_test.py	84
lingvo/core/ema_test.py	95
lingvo/core/entmax_test.py	39
lingvo/core/favor_attention_test.py	103
lingvo/core/flat_beam_search_helper_test.py	293
lingvo/core/generic_input_test.py	282
lingvo/core/gpipe_test.py	177
lingvo/core/graddrop_test.py	77
lingvo/core/gshard_builder_test.py	698
lingvo/core/gshard_layers_test.py	77
lingvo/core/hyperparams_test.py	598
lingvo/core/inference_graph_exporter_test.py	351
lingvo/core/input_generator_helper_test.py	170
lingvo/core/insertion_test.py	249
lingvo/core/inspect_utils_test.py	126
lingvo/core/layers_test.py	5328
lingvo/core/layers_with_attention_test.py	2554
lingvo/core/layers_with_gpipe_test.py	870
lingvo/core/learner_test.py	135
lingvo/core/lstm_frnn_layer_test.py	176
lingvo/core/matrix_functions_test.py	28
lingvo/core/metrics_test.py	144
lingvo/core/ml_perf_bleu_metric_test.py	11
lingvo/core/ml_perf_log_test.py	16
lingvo/core/multitask_model_test.py	97
lingvo/core/ops/assert_ops_test.py	37
lingvo/core/ops/beam_search_step_op_test.py	1000
lingvo/core/ops/beam_search_step_op_top_k_test.cc	126
lingvo/core/ops/best_step_op_test.py	55
lingvo/core/ops/functional_ops_test.py	26
lingvo/core/ops/input_common_test.cc	23
lingvo/core/ops/mass_op_test.py	846
lingvo/core/ops/pack_ops_test.py	430
lingvo/core/ops/preconditioner_op_kernels_test.py	64
lingvo/core/ops/random_ops_test.py	27
lingvo/core/ops/record_batcher_test.cc	259
lingvo/core/ops/record_yielder_test.cc	465
lingvo/core/ops/simple_vocab_test.py	137
lingvo/core/ops/static_map_op_test.py	35
lingvo/core/ops/text_packing_test.cc	216
lingvo/core/ops/tokenizer_ops_test.py	299
lingvo/core/ops/weighted_mix_record_yielder_test.cc	178
lingvo/core/optimizer_eager_test.py	140
lingvo/core/optimizer_test.py	174
lingvo/core/plot_test.py	229
lingvo/core/predictor_test.py	69
lingvo/core/py_utils_eager_test.py	37
lingvo/core/py_utils_test.py	3357
lingvo/core/quant_utils_test.py	310
lingvo/core/recurrent_test.py	648
lingvo/core/retry_test.py	52
lingvo/core/reversible_layers_test.py	138
lingvo/core/rnn_cell_test.py	1269
lingvo/core/rnn_layers_test.py	1287
lingvo/core/routing_config_helper_test.py	22
lingvo/core/saver_test.py	193
lingvo/core/schedule_test.py	664
lingvo/core/scorers_test.py	66
lingvo/core/self_attention_layer_test.py	246
lingvo/core/sendrecv_test.py	61
lingvo/core/spectrum_augmenter_on_device_test.py	461
lingvo/core/spectrum_augmenter_test.py	824
lingvo/core/step_test.py	143
lingvo/core/steps/attention_steps_test.py	512
lingvo/core/steps/embedding_steps_test.py	103
lingvo/core/steps/rnn_steps_test.py	189
lingvo/core/summary_utils_test.py	21
lingvo/core/symbolic_test.py	47
lingvo/core/target_sequence_sampler_test.py	375
lingvo/core/task_scheduler_test.py	136
lingvo/core/test_helper.py	4
lingvo/core/test_trainer_utils.py	42
lingvo/core/test_utils.py	125
lingvo/core/test_utils_test.py	28
lingvo/core/tokenizers_test.py	73
lingvo/core/tpu_summary_test.py	186
lingvo/core/tshape_test.py	19
lingvo/core/var_tmp_wrappers_test.py	26
lingvo/core/wpm_encoder_test.py	76
lingvo/core/wpm_tokenizer_test.py	43
lingvo/datasets_test.py	82
lingvo/jax/asserts_test.py	196
lingvo/jax/base_input_test.py	165
lingvo/jax/checkpoint_managers_test.py	442
lingvo/jax/layers/attentions_test.py	475
lingvo/jax/layers/augmentations_test.py	61
lingvo/jax/layers/conformers_test.py	118
lingvo/jax/layers/ctc_objectives_test.py	238
lingvo/jax/layers/embedding_softmax_test.py	417
lingvo/jax/layers/linears_test.py	279
lingvo/jax/layers/ngrammer_test.py	300
lingvo/jax/layers/normalizations_test.py	253
lingvo/jax/layers/poolings_test.py	117
lingvo/jax/layers/quantizer_objectives_test.py	21
lingvo/jax/layers/quantizer_test.py	64
lingvo/jax/layers/recurrent_test.py	205
lingvo/jax/layers/spectrum_augmenter_test.py	84
lingvo/jax/layers/stats_test.py	18
lingvo/jax/layers/stochastics_test.py	69
lingvo/jax/layers/test_layers.py	194
lingvo/jax/layers/transformers_test.py	1448
lingvo/jax/learners_test.py	381
lingvo/jax/model_imports_test.py	5
lingvo/jax/model_registry_test.py	26
lingvo/jax/model_test.py	252
lingvo/jax/py_utils_test.py	30
lingvo/jax/schedules_test.py	356
lingvo/jax/tasks/lm/input_generator_test.py	70
lingvo/jax/tasks/test/__init__.py	1
lingvo/jax/tasks/test/params/params.py	1
lingvo/jax/tasks/test/params/synthetic.py	10
lingvo/jax/test_utils.py	220
lingvo/model_import_test.py	5
lingvo/model_registry_test.py	200
lingvo/models_test.py	14
lingvo/tasks/asr/decoder_test.py	604
lingvo/tasks/asr/decoder_utils_test.py	144
lingvo/tasks/asr/encoder_test.py	289
lingvo/tasks/asr/eos_normalization_test.py	124
lingvo/tasks/asr/frontend_test.py	393
lingvo/tasks/asr/input_generator_test.py	122
lingvo/tasks/asr/model_test.py	519
lingvo/tasks/asr/tools/simple_wer_test.py	12
lingvo/tasks/asr/tools/simple_wer_v2_test.py	60
lingvo/tasks/car/breakdown_metric_test.py	319
lingvo/tasks/car/builder_lib_test.py	418
lingvo/tasks/car/calibration_processing_test.py	103
lingvo/tasks/car/car_layers_test.py	58
lingvo/tasks/car/car_lib_test.py	371
lingvo/tasks/car/detection_3d_lib_test.py	415
lingvo/tasks/car/detection_3d_metrics_test.py	70
lingvo/tasks/car/detection_decoder_test.py	82
lingvo/tasks/car/geometry_test.py	415
lingvo/tasks/car/input_extractor_test.py	108
lingvo/tasks/car/input_preprocessors_test.py	109
lingvo/tasks/car/lr_util_test.py	58
lingvo/tasks/car/ops/average_precision_3d_op_test.py	90
lingvo/tasks/car/ops/nms_3d_op_test.py	158
lingvo/tasks/car/ops/point_grid_op_test.py	91
lingvo/tasks/car/ops/ps_utils_test.cc	292
lingvo/tasks/car/ops/sampling_ops_test.py	118
lingvo/tasks/car/pointnet_test.py	46
lingvo/tasks/car/summary_test.py	69
lingvo/tasks/car/tools/kitti_data_test.py	89
lingvo/tasks/car/transform_util_test.py	74
lingvo/tasks/car/waymo/waymo_ap_metric_test.py	87
lingvo/tasks/car/waymo/waymo_open_input_generator_test.py	26
lingvo/tasks/image/classifier_test.py	95
lingvo/tasks/image/input_generator_test.py	68
lingvo/tasks/lm/input_generator_test.py	34
lingvo/tasks/lm/layers_test.py	823
lingvo/tasks/lm/model_test.py	135
lingvo/tasks/milan/dataset_spec_test.py	24
lingvo/tasks/milan/dual_encoder_test.py	79
lingvo/tasks/milan/image_preprocessor_test.py	52
lingvo/tasks/milan/input_generator_test.py	25
lingvo/tasks/milan/labels_test.py	143
lingvo/tasks/milan/params/generic_datasets_test.py	28
lingvo/tasks/milan/score_functions_test.py	12
lingvo/tasks/milan/tf_hub_layers_test.py	102
lingvo/tasks/milan/tpu_utils_test.py	31
lingvo/tasks/milan/transformers_test.py	22
lingvo/tasks/milan/utils_test.py	54
lingvo/tasks/mt/data_augmenter_test.py	42
lingvo/tasks/mt/decoder_test.py	1635
lingvo/tasks/mt/encoder_test.py	590
lingvo/tasks/mt/input_generator_test.py	548
lingvo/tasks/mt/layers_test.py	284
lingvo/tasks/mt/model_test.py	839
lingvo/tasks/punctuator/input_generator_test.py	53
lingvo/tasks/punctuator/model_test.py	111
lingvo/tools/audio_lib_test.py	55
lingvo/tools/beam_utils_test.py	37
lingvo/tools/compare_params_test.py	19
lingvo/trainer_test.py	365