file 1	file 2	# same commits	# commits file 1	# commits file 2
lingvo/core/py_utils.py	lingvo/core/base_model.py	18	451	213
lingvo/jax/train.py	lingvo/jax/eval.py	17	24	32
lingvo/core/base_model.py	lingvo/executor.py	15	213	90
lingvo/jax/layers/attentions.py	lingvo/jax/layers/transformers.py	15	21	47
lingvo/core/base_layer.py	lingvo/core/base_model.py	13	132	213
lingvo/trainer.py	lingvo/base_runner.py	13	248	84
lingvo/core/ops/beam_search_step_op_kernels.cc	lingvo/core/ops/x_ops.cc	13	46	55
lingvo/core/program.py	lingvo/executor.py	12	136	90
lingvo/core/program.py	lingvo/core/base_model.py	12	136	213
lingvo/core/rnn_cell.py	lingvo/core/layers.py	12	118	286
lingvo/core/gshard_layers.py	lingvo/core/gshard_builder.py	11	52	86
lingvo/jax/layers/transformers.py	lingvo/jax/layers/__init__.py	10	47	21
lingvo/jax/trainer_lib.py	lingvo/jax/eval.py	10	12	32
lingvo/core/base_layer.py	lingvo/core/py_utils.py	10	132	451
lingvo/core/attention.py	lingvo/core/layers.py	10	107	286
lingvo/core/checkpointer.py	lingvo/core/base_model.py	9	51	213
lingvo/core/checkpointer.py	lingvo/executor.py	9	51	90
lingvo/jax/layers/embedding_softmax.py	lingvo/jax/layers/transformers.py	9	11	47
lingvo/jax/train.py	lingvo/jax/model_utils.py	9	24	9
lingvo/core/layers.py	lingvo/core/batch_major_attention.py	9	286	212
lingvo/core/attention.py	lingvo/core/rnn_cell.py	9	107	118
lingvo/jax/eval.py	lingvo/jax/py_utils.py	8	32	12
lingvo/jax/eval.py	lingvo/jax/model.py	8	32	24
lingvo/jax/model_utils.py	lingvo/jax/eval.py	8	9	32
lingvo/core/py_utils.py	lingvo/executor.py	8	451	90
lingvo/jax/main.py	lingvo/jax/eval.py	8	14	32
lingvo/core/layers.py	lingvo/core/py_utils.py	8	286	451
lingvo/core/base_layer.py	lingvo/executor.py	8	132	90
lingvo/trainer.py	lingvo/executor.py	8	248	90
lingvo/jax/train.py	lingvo/jax/py_utils.py	7	24	12
lingvo/jax/train.py	lingvo/jax/trainer_lib.py	7	24	12
lingvo/core/py_utils.py	lingvo/core/program.py	7	451	136
lingvo/core/gshard_layers.py	lingvo/core/batch_major_attention.py	7	52	212
lingvo/core/tpu_embedding_layers.py	lingvo/core/base_model.py	7	42	213
lingvo/core/base_layer.py	lingvo/core/layers.py	7	132	286
lingvo/core/attention.py	lingvo/core/quant_utils.py	7	107	57
lingvo/trainer.py	lingvo/core/base_model.py	7	248	213
lingvo/core/ops/beam_search_step_op_kernels.h	lingvo/core/ops/beam_search_step_op_kernels.cc	7	16	46
lingvo/jax/layers/attentions.py	lingvo/jax/layers/__init__.py	6	21	21
lingvo/jax/layers/embedding_softmax.py	lingvo/jax/layers/attentions.py	6	11	21
lingvo/jax/layers/embedding_softmax.py	lingvo/jax/layers/__init__.py	6	11	21
lingvo/jax/model_utils.py	lingvo/jax/py_utils.py	6	9	12
lingvo/core/py_utils.py	lingvo/core/optimizer.py	6	451	56
lingvo/jax/main.py	lingvo/jax/train.py	6	14	24
lingvo/core/gshard_layers.py	lingvo/core/py_utils.py	6	52	451
lingvo/core/tpu_embedding_layers.py	lingvo/core/py_utils.py	6	42	451
lingvo/base_runner.py	lingvo/executor.py	6	84	90
lingvo/core/base_layer.py	lingvo/core/program.py	6	132	136
lingvo/core/rnn_cell.py	lingvo/core/py_utils.py	6	118	451
lingvo/core/learner.py	lingvo/core/base_model.py	6	52	213
lingvo/core/learner.py	lingvo/core/py_utils.py	6	52	451
lingvo/trainer.py	lingvo/core/program.py	6	248	136
pip_package/setup.py	pip_package/build.Dockerfile	6	28	14
lingvo/core/quant_test_lib.py	lingvo/core/quant_utils.py	6	10	57
lingvo/jax/model.py	lingvo/jax/layers/transformers.py	5	24	47
lingvo/jax/optimizers.py	lingvo/jax/learners.py	5	32	8
lingvo/jax/optimizers.py	lingvo/jax/model.py	5	32	24
lingvo/jax/eval.py	lingvo/jax/checkpoints.py	5	32	13
lingvo/jax/trainer_lib.py	lingvo/jax/model.py	5	12	24
lingvo/jax/model_utils.py	lingvo/jax/trainer_lib.py	5	9	12
lingvo/jax/train.py	lingvo/jax/checkpoints.py	5	24	13
lingvo/jax/base_layer.py	lingvo/jax/trainer_lib.py	5	10	12
lingvo/jax/base_layer.py	lingvo/jax/eval.py	5	10	32
lingvo/jax/base_layer.py	lingvo/jax/train.py	5	10	24
lingvo/jax/summary_utils.py	lingvo/jax/eval.py	5	7	32
lingvo/jax/summary_utils.py	lingvo/jax/model_utils.py	5	7	9
lingvo/jax/summary_utils.py	lingvo/jax/train.py	5	7	24
lingvo/core/base_input_generator.py	lingvo/core/program.py	5	176	136
lingvo/core/base_input_generator.py	lingvo/core/datasource.py	5	176	55
lingvo/core/tpu_embedding_layers.py	lingvo/core/base_input_generator.py	5	42	176
lingvo/jax/layers/repeats.py	lingvo/jax/layers/transformers.py	5	5	47
lingvo/core/base_layer.py	lingvo/core/gshard_layers.py	5	132	52
lingvo/core/rnn_cell.py	lingvo/core/quant_utils.py	5	118	57
lingvo/core/builder_layers.py	lingvo/core/gshard_builder.py	5	52	86
lingvo/core/rnn_layers.py	lingvo/core/base_layer.py	5	72	132
lingvo/core/rnn_layers.py	lingvo/core/rnn_cell.py	5	72	118
lingvo/trainer.py	lingvo/core/base_input_generator.py	5	248	176
lingvo/core/quant_test_lib.py	lingvo/core/rnn_cell.py	5	10	118
lingvo/core/quant_test_lib.py	lingvo/core/attention.py	5	10	107
lingvo/core/beam_search_helper.py	lingvo/core/ops/x_ops.cc	5	71	55
lingvo/core/beam_search_helper.py	lingvo/core/ops/beam_search_step_op_kernels.cc	5	71	46
lingvo/core/program.py	lingvo/core/checkpointer.py	4	136	51
lingvo/jax/py_utils.py	lingvo/jax/checkpoints.py	4	12	13
lingvo/model_registry.py	lingvo/core/program.py	4	31	136
lingvo/jax/eval.py	lingvo/jax/base_input.py	4	32	14
lingvo/jax/trainer_lib.py	lingvo/jax/py_utils.py	4	12	12
lingvo/jax/train.py	lingvo/jax/model.py	4	24	24
lingvo/jax/train.py	lingvo/jax/base_input.py	4	24	14
lingvo/jax/layers/convolutions.py	lingvo/jax/layers/__init__.py	4	8	21
lingvo/jax/layers/resnets.py	lingvo/jax/layers/convolutions.py	4	4	8
lingvo/jax/layers/normalizations.py	lingvo/jax/layers/transformers.py	4	8	47
lingvo/jax/layers/normalizations.py	lingvo/jax/layers/__init__.py	4	8	21
lingvo/jax/layers/normalizations.py	lingvo/jax/layers/convolutions.py	4	8	8
lingvo/jax/layers/recurrent.py	lingvo/jax/base_layer.py	4	6	10
lingvo/core/py_utils.py	lingvo/core/batch_major_attention.py	4	451	212
lingvo/jax/main.py	lingvo/jax/base_layer.py	4	14	10
lingvo/jax/main.py	lingvo/jax/trainer_lib.py	4	14	12
lingvo/jax/main.py	lingvo/jax/checkpoints.py	4	14	13
lingvo/jax/main.py	lingvo/jax/py_utils.py	4	14	12
lingvo/jax/summary_utils.py	lingvo/jax/py_utils.py	4	7	12
lingvo/jax/summary_utils.py	lingvo/jax/trainer_lib.py	4	7	12
lingvo/jax/layers/linears.py	lingvo/jax/layers/transformers.py	4	7	47
lingvo/jax/layers/linears.py	lingvo/jax/layers/attentions.py	4	7	21
lingvo/core/gshard_layers.py	lingvo/core/base_model.py	4	52	213
lingvo/core/quant_utils.py	lingvo/core/layers.py	4	57	286
lingvo/base_runner.py	lingvo/core/base_model.py	4	84	213
lingvo/core/base_layer.py	lingvo/core/batch_major_attention.py	4	132	212
lingvo/core/cluster.py	lingvo/core/program.py	4	57	136
lingvo/core/cluster.py	lingvo/core/py_utils.py	4	57	451
lingvo/eager_runners.py	lingvo/executor.py	4	7	90
lingvo/runners.py	lingvo/executor.py	4	6	90
lingvo/runners.py	lingvo/eager_runners.py	4	6	7
lingvo/core/rnn_cell.py	lingvo/core/base_layer.py	4	118	132
lingvo/core/conformer_layer.py	lingvo/core/gshard_layers.py	4	55	52
lingvo/core/conformer_layer.py	lingvo/core/batch_major_attention.py	4	55	212
lingvo/core/rnn_layers.py	lingvo/core/base_model.py	4	72	213
lingvo/core/rnn_layers.py	lingvo/core/layers.py	4	72	286
lingvo/core/rnn_layers.py	lingvo/core/py_utils.py	4	72	451
lingvo/core/attention.py	lingvo/core/py_utils.py	4	107	451
lingvo/jax/gshard_utils.py	lingvo/jax/layers/transformers.py	4	4	47
lingvo/jax/tasks/lm/params/bert.py	lingvo/jax/tasks/lm/input_generator.py	4	8	8
lingvo/trainer_utils.py	lingvo/trainer.py	4	4	248
lingvo/jax/layers/activations.py	lingvo/jax/layers/__init__.py	4	4	21
lingvo/jax/layers/activations.py	lingvo/jax/layers/convolutions.py	4	4	8
lingvo/core/quant_test_lib.py	lingvo/core/layers.py	4	10	286
lingvo/core/ops/beam_search_step_op_kernels.h	lingvo/core/ops/x_ops.cc	4	16	55
lingvo/core/layers_with_attention.py	lingvo/core/batch_major_attention.py	4	87	212
lingvo/tasks/asr/metrics_calculator.py	lingvo/tasks/asr/decoder_metrics.py	4	4	14
lingvo/core/early_stop.py	lingvo/base_runner.py	4	14	84
lingvo/core/saver.py	lingvo/core/base_model.py	3	11	213
lingvo/core/saver.py	lingvo/core/checkpointer.py	3	11	51
lingvo/jax/py_utils.py	lingvo/jax/layers/__init__.py	3	12	21
lingvo/jax/model.py	lingvo/jax/py_utils.py	3	24	12
lingvo/jax/model.py	lingvo/jax/base_input.py	3	24	14
lingvo/jax/model.py	lingvo/jax/checkpoints.py	3	24	13
lingvo/jax/model.py	lingvo/jax/learners.py	3	24	8
lingvo/jax/model.py	lingvo/jax/layers/__init__.py	3	24	21
lingvo/core/datasource.py	lingvo/core/program.py	3	55	136
lingvo/jax/layers/attentions.py	lingvo/jax/model.py	3	21	24
lingvo/jax/optimizers.py	lingvo/jax/layers/__init__.py	3	32	21
lingvo/jax/optimizers.py	lingvo/jax/layers/transformers.py	3	32	47
lingvo/jax/optimizers.py	lingvo/jax/layers/attentions.py	3	32	21
lingvo/jax/optimizers.py	lingvo/jax/layers/embedding_softmax.py	3	32	11
lingvo/jax/eval.py	lingvo/jax/layers/__init__.py	3	32	21
lingvo/jax/eval.py	lingvo/jax/layers/transformers.py	3	32	47
lingvo/jax/trainer_lib.py	lingvo/jax/optimizers.py	3	12	32
lingvo/jax/tasks/lm/model_params.py	lingvo/jax/layers/transformers.py	3	8	47
lingvo/jax/model_utils.py	lingvo/jax/checkpoints.py	3	9	13
lingvo/jax/model_utils.py	lingvo/jax/base_input.py	3	9	14
lingvo/jax/model_utils.py	lingvo/jax/model.py	3	9	24
lingvo/core/batch_major_attention.py	lingvo/core/base_model.py	3	212	213
lingvo/jax/layers/convolutions.py	lingvo/jax/model.py	3	8	24
lingvo/jax/layers/convolutions.py	lingvo/jax/layers/transformers.py	3	8	47
lingvo/jax/layers/convolutions.py	lingvo/jax/layers/attentions.py	3	8	21
lingvo/jax/layers/convolutions.py	lingvo/jax/layers/embedding_softmax.py	3	8	11
lingvo/jax/layers/convolutions.py	lingvo/jax/optimizers.py	3	8	32
lingvo/jax/layers/resnets.py	lingvo/jax/model.py	3	4	24
lingvo/jax/layers/resnets.py	lingvo/jax/layers/__init__.py	3	4	21
lingvo/jax/layers/resnets.py	lingvo/jax/layers/transformers.py	3	4	47
lingvo/jax/layers/resnets.py	lingvo/jax/layers/attentions.py	3	4	21
lingvo/jax/layers/resnets.py	lingvo/jax/layers/embedding_softmax.py	3	4	11
lingvo/jax/layers/resnets.py	lingvo/jax/optimizers.py	3	4	32
lingvo/jax/base_layer.py	lingvo/jax/layers/transformers.py	3	10	47
lingvo/jax/base_layer.py	lingvo/jax/model_utils.py	3	10	9
lingvo/jax/layers/normalizations.py	lingvo/jax/trainer_lib.py	3	8	12
lingvo/jax/layers/normalizations.py	lingvo/jax/base_layer.py	3	8	10
lingvo/jax/layers/normalizations.py	lingvo/jax/layers/attentions.py	3	8	21
lingvo/jax/layers/normalizations.py	lingvo/jax/layers/embedding_softmax.py	3	8	11
lingvo/jax/layers/normalizations.py	lingvo/jax/optimizers.py	3	8	32
lingvo/jax/layers/normalizations.py	lingvo/jax/layers/resnets.py	3	8	4
lingvo/jax/layers/recurrent.py	lingvo/jax/trainer_lib.py	3	6	12
lingvo/jax/layers/recurrent.py	lingvo/jax/layers/normalizations.py	3	6	8
lingvo/jax/layers/recurrent.py	lingvo/jax/layers/transformers.py	3	6	47
lingvo/core/py_utils.py	lingvo/core/checkpointer.py	3	451	51
lingvo/jax/main.py	lingvo/jax/model.py	3	14	24
lingvo/jax/main.py	lingvo/jax/model_utils.py	3	14	9
lingvo/core/layers.py	lingvo/core/base_model.py	3	286	213
lingvo/core/gshard_builder.py	lingvo/core/layers.py	3	86	286
lingvo/jax/summary_utils.py	lingvo/jax/layers/transformers.py	3	7	47
lingvo/jax/summary_utils.py	lingvo/jax/checkpoints.py	3	7	13
lingvo/jax/summary_utils.py	lingvo/jax/main.py	3	7	14
lingvo/jax/summary_utils.py	lingvo/jax/base_layer.py	3	7	10
lingvo/jax/layers/linears.py	lingvo/jax/layers/__init__.py	3	7	21
lingvo/jax/layers/linears.py	lingvo/jax/layers/embedding_softmax.py	3	7	11
lingvo/jax/layers/linears.py	lingvo/jax/optimizers.py	3	7	32
lingvo/jax/layers/linears.py	lingvo/jax/layers/convolutions.py	3	7	8
lingvo/jax/layers/linears.py	lingvo/jax/layers/resnets.py	3	7	4
lingvo/jax/layers/linears.py	lingvo/jax/layers/normalizations.py	3	7	8
lingvo/core/base_input_generator.py	lingvo/executor.py	3	176	90
lingvo/core/base_input_generator.py	lingvo/core/base_model.py	3	176	213
lingvo/jax/layers/repeats.py	lingvo/jax/summary_utils.py	3	5	7
lingvo/jax/layers/repeats.py	lingvo/jax/base_layer.py	3	5	10
lingvo/jax/layers/repeats.py	lingvo/jax/layers/recurrent.py	3	5	6
lingvo/jax/layers/repeats.py	lingvo/jax/layers/__init__.py	3	5	21
lingvo/jax/layers/repeats.py	lingvo/jax/layers/attentions.py	3	5	21
lingvo/jax/layers/repeats.py	lingvo/jax/layers/embedding_softmax.py	3	5	11
lingvo/jax/layers/repeats.py	lingvo/jax/optimizers.py	3	5	32
lingvo/jax/layers/repeats.py	lingvo/jax/layers/convolutions.py	3	5	8
lingvo/jax/layers/repeats.py	lingvo/jax/layers/resnets.py	3	5	4
lingvo/jax/layers/repeats.py	lingvo/jax/layers/normalizations.py	3	5	8
lingvo/jax/layers/repeats.py	lingvo/jax/layers/linears.py	3	5	7
lingvo/jax/tasks/lm/params/lm_cloud.py	lingvo/jax/tasks/lm/model_params.py	3	7	8
lingvo/core/base_layer.py	lingvo/core/checkpointer.py	3	132	51
lingvo/core/cluster.py	lingvo/core/base_model.py	3	57	213
lingvo/core/cluster.py	lingvo/core/datasource.py	3	57	55
lingvo/eager_runners.py	lingvo/base_runner.py	3	7	84
lingvo/runners.py	lingvo/base_runner.py	3	6	84
lingvo/core/rnn_cell.py	lingvo/core/base_model.py	3	118	213
lingvo/core/rnn_cell.py	lingvo/core/batch_major_attention.py	3	118	212
lingvo/core/builder_layers.py	lingvo/core/gshard_layers.py	3	52	52
lingvo/core/conformer_layer.py	lingvo/core/gshard_builder.py	3	55	86
lingvo/core/rnn_layers.py	lingvo/tasks/mt/decoder.py	3	72	120
lingvo/core/rnn_layers.py	lingvo/core/gshard_layers.py	3	72	52
lingvo/core/rnn_layers.py	lingvo/core/batch_major_attention.py	3	72	212
lingvo/jax/pip_package/setup.py	lingvo/jax/pip_package/build.Dockerfile	3	4	3
lingvo/tasks/mt/encoder.py	lingvo/tasks/mt/decoder.py	3	48	120
lingvo/core/attention.py	lingvo/core/rnn_layers.py	3	107	72
lingvo/core/learner.py	lingvo/core/batch_major_attention.py	3	52	212
lingvo/core/learner.py	lingvo/core/layers.py	3	52	286
lingvo/core/learner.py	lingvo/core/base_layer.py	3	52	132
lingvo/core/learner.py	lingvo/core/rnn_layers.py	3	52	72
lingvo/jax/tasks/lm/input_generator.py	lingvo/jax/base_input.py	3	8	14
lingvo/jax/tasks/lm/input_generator.py	lingvo/jax/py_utils.py	3	8	12
lingvo/jax/tasks/lm/input_generator.py	lingvo/jax/eval.py	3	8	32
lingvo/jax/tasks/lm/input_generator.py	lingvo/jax/model_utils.py	3	8	9
lingvo/jax/tasks/lm/input_generator.py	lingvo/jax/train.py	3	8	24
lingvo/core/builder.py	lingvo/core/builder_layers.py	3	10	52
lingvo/jax/tasks/lm/params/bert.py	lingvo/jax/layers/__init__.py	3	8	21
lingvo/jax/tasks/lm/params/bert.py	lingvo/jax/layers/transformers.py	3	8	47
lingvo/jax/tasks/lm/params/bert.py	lingvo/jax/layers/attentions.py	3	8	21
lingvo/jax/tasks/lm/params/bert.py	lingvo/jax/layers/embedding_softmax.py	3	8	11
lingvo/jax/tasks/lm/params/bert.py	lingvo/jax/optimizers.py	3	8	32
lingvo/jax/tasks/lm/params/bert.py	lingvo/jax/layers/convolutions.py	3	8	8
lingvo/jax/tasks/lm/params/bert.py	lingvo/jax/layers/resnets.py	3	8	4
lingvo/jax/tasks/lm/params/bert.py	lingvo/jax/layers/normalizations.py	3	8	8
lingvo/jax/tasks/lm/params/bert.py	lingvo/jax/layers/linears.py	3	8	7
lingvo/jax/tasks/lm/params/bert.py	lingvo/jax/layers/repeats.py	3	8	5
lingvo/jax/tasks/lm/params/bert.py	lingvo/jax/base_input.py	3	8	14
lingvo/jax/tasks/lm/params/bert.py	lingvo/jax/eval.py	3	8	32
lingvo/jax/tasks/lm/params/bert.py	lingvo/jax/train.py	3	8	24
lingvo/trainer.py	lingvo/eager_runners.py	3	248	7
lingvo/trainer.py	lingvo/runners.py	3	248	6
lingvo/trainer.py	lingvo/core/py_utils.py	3	248	451
lingvo/trainer.py	lingvo/core/cluster.py	3	248	57
lingvo/trainer_utils.py	lingvo/base_runner.py	3	4	84
lingvo/jax/base_model_params.py	lingvo/jax/eval.py	3	4	32
lingvo/jax/base_model_params.py	lingvo/jax/train.py	3	4	24
lingvo/jax/base_model_params.py	lingvo/jax/tasks/lm/params/bert.py	3	4	8
lingvo/jax/schedules.py	lingvo/jax/layers/__init__.py	3	4	21
lingvo/jax/schedules.py	lingvo/jax/layers/transformers.py	3	4	47
lingvo/jax/schedules.py	lingvo/jax/layers/attentions.py	3	4	21
lingvo/jax/schedules.py	lingvo/jax/layers/embedding_softmax.py	3	4	11
lingvo/jax/schedules.py	lingvo/jax/optimizers.py	3	4	32
lingvo/jax/schedules.py	lingvo/jax/layers/convolutions.py	3	4	8
lingvo/jax/schedules.py	lingvo/jax/layers/resnets.py	3	4	4
lingvo/jax/schedules.py	lingvo/jax/layers/normalizations.py	3	4	8
lingvo/jax/schedules.py	lingvo/jax/layers/linears.py	3	4	7
lingvo/jax/schedules.py	lingvo/jax/layers/repeats.py	3	4	5
lingvo/jax/schedules.py	lingvo/jax/tasks/lm/params/bert.py	3	4	8
lingvo/core/ops/input_common.h	lingvo/core/ops/input_common.cc	3	16	12
lingvo/jax/layers/activations.py	lingvo/jax/layers/transformers.py	3	4	47
lingvo/jax/layers/activations.py	lingvo/jax/layers/attentions.py	3	4	21
lingvo/jax/layers/activations.py	lingvo/jax/layers/embedding_softmax.py	3	4	11
lingvo/jax/layers/activations.py	lingvo/jax/optimizers.py	3	4	32
lingvo/jax/layers/activations.py	lingvo/jax/layers/resnets.py	3	4	4
lingvo/jax/layers/activations.py	lingvo/jax/layers/normalizations.py	3	4	8
lingvo/jax/layers/activations.py	lingvo/jax/layers/linears.py	3	4	7
lingvo/jax/layers/activations.py	lingvo/jax/layers/repeats.py	3	4	5
lingvo/jax/layers/activations.py	lingvo/jax/tasks/lm/params/bert.py	3	4	8
lingvo/jax/layers/activations.py	lingvo/jax/schedules.py	3	4	4
lingvo/core/summary_utils.py	lingvo/trainer.py	3	49	248
lingvo/jax/layers/poolings.py	lingvo/jax/layers/__init__.py	3	3	21
lingvo/jax/layers/poolings.py	lingvo/jax/layers/transformers.py	3	3	47
lingvo/jax/layers/poolings.py	lingvo/jax/layers/attentions.py	3	3	21
lingvo/jax/layers/poolings.py	lingvo/jax/layers/embedding_softmax.py	3	3	11
lingvo/jax/layers/poolings.py	lingvo/jax/optimizers.py	3	3	32
lingvo/jax/layers/poolings.py	lingvo/jax/layers/convolutions.py	3	3	8
lingvo/jax/layers/poolings.py	lingvo/jax/layers/resnets.py	3	3	4
lingvo/jax/layers/poolings.py	lingvo/jax/layers/normalizations.py	3	3	8
lingvo/jax/layers/poolings.py	lingvo/jax/layers/linears.py	3	3	7
lingvo/jax/layers/poolings.py	lingvo/jax/layers/repeats.py	3	3	5
lingvo/jax/layers/poolings.py	lingvo/jax/tasks/lm/params/bert.py	3	3	8
lingvo/jax/layers/poolings.py	lingvo/jax/schedules.py	3	3	4
lingvo/jax/layers/poolings.py	lingvo/jax/layers/activations.py	3	3	4
lingvo/jax/layers/stochastics.py	lingvo/jax/layers/__init__.py	3	3	21
lingvo/jax/layers/stochastics.py	lingvo/jax/layers/transformers.py	3	3	47
lingvo/jax/layers/stochastics.py	lingvo/jax/layers/attentions.py	3	3	21
lingvo/jax/layers/stochastics.py	lingvo/jax/layers/embedding_softmax.py	3	3	11
lingvo/jax/layers/stochastics.py	lingvo/jax/optimizers.py	3	3	32
lingvo/jax/layers/stochastics.py	lingvo/jax/layers/convolutions.py	3	3	8
lingvo/jax/layers/stochastics.py	lingvo/jax/layers/resnets.py	3	3	4
lingvo/jax/layers/stochastics.py	lingvo/jax/layers/normalizations.py	3	3	8
lingvo/jax/layers/stochastics.py	lingvo/jax/layers/linears.py	3	3	7
lingvo/jax/layers/stochastics.py	lingvo/jax/layers/repeats.py	3	3	5
lingvo/jax/layers/stochastics.py	lingvo/jax/tasks/lm/params/bert.py	3	3	8
lingvo/jax/layers/stochastics.py	lingvo/jax/schedules.py	3	3	4
lingvo/jax/layers/stochastics.py	lingvo/jax/layers/activations.py	3	3	4
lingvo/jax/layers/stochastics.py	lingvo/jax/layers/poolings.py	3	3	3
lingvo/tasks/lm/params/wiki_bert.py	lingvo/core/program.py	3	8	136
lingvo/core/bn_layers.py	lingvo/core/batch_major_attention.py	3	72	212
lingvo/core/bn_layers.py	lingvo/core/py_utils.py	3	72	451
lingvo/core/bn_layers.py	lingvo/core/layers.py	3	72	286
lingvo/core/bn_layers.py	lingvo/core/rnn_cell.py	3	72	118
lingvo/core/bn_layers.py	lingvo/core/attention.py	3	72	107
lingvo/tasks/lm/input_generator.py	lingvo/core/layers.py	3	24	286
lingvo/tasks/lm/input_generator.py	lingvo/core/gshard_builder.py	3	24	86
lingvo/core/hyperparams.py	lingvo/core/batch_major_attention.py	3	64	212
lingvo/core/hyperparams.py	lingvo/core/py_utils.py	3	64	451
lingvo/core/hyperparams.py	lingvo/core/base_layer.py	3	64	132
lingvo/core/ops/__init__.py	lingvo/core/ops/x_ops.cc	3	15	55
lingvo/core/pruning_utils.py	lingvo/core/rnn_cell.py	3	5	118
lingvo/jax/layers/__init__.py	lingvo/jax/layers/conformers.py	2	21	5
lingvo/jax/layers/__init__.py	lingvo/jax/checkpoints.py	2	21	13
lingvo/jax/layers/pipeline.py	lingvo/jax/layers/__init__.py	2	5	21
lingvo/jax/layers/transformers.py	lingvo/jax/layers/pipeline.py	2	47	5
lingvo/jax/layers/transformers.py	lingvo/jax/base_input.py	2	47	14
lingvo/jax/layers/transformers.py	lingvo/jax/layers/conformers.py	2	47	5
lingvo/jax/layers/transformers.py	lingvo/jax/checkpoints.py	2	47	13
lingvo/jax/py_utils.py	lingvo/jax/base_input.py	2	12	14
lingvo/jax/py_utils.py	lingvo/jax/layers/transformers.py	2	12	47
lingvo/core/metrics.py	lingvo/core/program.py	2	38	136
lingvo/tasks/mt/decoder.py	lingvo/core/base_model.py	2	120	213
lingvo/jax/layers/attentions.py	lingvo/jax/checkpoints.py	2	21	13
lingvo/jax/layers/attentions.py	lingvo/jax/py_utils.py	2	21	12
lingvo/jax/layers/embedding_softmax.py	lingvo/jax/checkpoints.py	2	11	13
lingvo/jax/layers/embedding_softmax.py	lingvo/jax/py_utils.py	2	11	12
lingvo/jax/layers/embedding_softmax.py	lingvo/jax/model.py	2	11	24
lingvo/jax/optimizers.py	lingvo/jax/checkpoints.py	2	32	13
lingvo/jax/optimizers.py	lingvo/jax/py_utils.py	2	32	12
lingvo/core/optimizer.py	lingvo/core/base_model.py	2	56	213
lingvo/core/optimizer.py	lingvo/core/checkpointer.py	2	56	51
lingvo/core/optimizer.py	lingvo/core/metrics.py	2	56	38
lingvo/jax/eval.py	lingvo/jax/layers/attentions.py	2	32	21
lingvo/jax/eval.py	lingvo/jax/layers/embedding_softmax.py	2	32	11
lingvo/jax/eval.py	lingvo/jax/optimizers.py	2	32	32
lingvo/jax/trainer_lib.py	lingvo/jax/checkpoints.py	2	12	13
lingvo/jax/trainer_lib.py	lingvo/jax/layers/__init__.py	2	12	21
lingvo/jax/trainer_lib.py	lingvo/jax/layers/transformers.py	2	12	47
lingvo/jax/trainer_lib.py	lingvo/jax/layers/attentions.py	2	12	21
lingvo/jax/trainer_lib.py	lingvo/jax/layers/embedding_softmax.py	2	12	11
lingvo/jax/model_utils.py	lingvo/jax/layers/__init__.py	2	9	21
lingvo/jax/model_utils.py	lingvo/jax/layers/transformers.py	2	9	47
lingvo/jax/model_utils.py	lingvo/jax/layers/attentions.py	2	9	21
lingvo/jax/model_utils.py	lingvo/jax/layers/embedding_softmax.py	2	9	11
lingvo/jax/model_utils.py	lingvo/jax/optimizers.py	2	9	32
lingvo/jax/train.py	lingvo/jax/layers/__init__.py	2	24	21
lingvo/jax/train.py	lingvo/jax/layers/transformers.py	2	24	47
lingvo/jax/train.py	lingvo/jax/layers/attentions.py	2	24	21
lingvo/jax/train.py	lingvo/jax/layers/embedding_softmax.py	2	24	11
lingvo/jax/train.py	lingvo/jax/optimizers.py	2	24	32
lingvo/core/batch_major_attention.py	lingvo/tasks/mt/decoder.py	2	212	120
lingvo/core/batch_major_attention.py	lingvo/core/saver.py	2	212	11
lingvo/jax/layers/convolutions.py	lingvo/jax/checkpoints.py	2	8	13
lingvo/jax/layers/convolutions.py	lingvo/jax/py_utils.py	2	8	12
lingvo/jax/layers/convolutions.py	lingvo/jax/eval.py	2	8	32
lingvo/jax/layers/convolutions.py	lingvo/jax/trainer_lib.py	2	8	12
lingvo/jax/layers/convolutions.py	lingvo/jax/model_utils.py	2	8	9
lingvo/jax/layers/convolutions.py	lingvo/jax/train.py	2	8	24
lingvo/jax/layers/resnets.py	lingvo/jax/checkpoints.py	2	4	13
lingvo/jax/layers/resnets.py	lingvo/jax/py_utils.py	2	4	12
lingvo/jax/layers/resnets.py	lingvo/jax/eval.py	2	4	32
lingvo/jax/layers/resnets.py	lingvo/jax/trainer_lib.py	2	4	12
lingvo/jax/layers/resnets.py	lingvo/jax/model_utils.py	2	4	9
lingvo/jax/layers/resnets.py	lingvo/jax/train.py	2	4	24
lingvo/jax/base_layer.py	lingvo/jax/layers/pipeline.py	2	10	5
lingvo/jax/base_layer.py	lingvo/jax/checkpoints.py	2	10	13
lingvo/jax/base_layer.py	lingvo/jax/layers/__init__.py	2	10	21
lingvo/jax/base_layer.py	lingvo/jax/py_utils.py	2	10	12
lingvo/jax/base_layer.py	lingvo/jax/model.py	2	10	24
lingvo/jax/base_layer.py	lingvo/jax/layers/attentions.py	2	10	21
lingvo/jax/base_layer.py	lingvo/jax/layers/embedding_softmax.py	2	10	11
lingvo/jax/base_layer.py	lingvo/jax/optimizers.py	2	10	32
lingvo/jax/base_layer.py	lingvo/jax/layers/convolutions.py	2	10	8
lingvo/jax/base_layer.py	lingvo/jax/layers/resnets.py	2	10	4
lingvo/jax/layers/ngrammer.py	lingvo/jax/layers/transformers.py	2	8	47
lingvo/jax/layers/normalizations.py	lingvo/jax/checkpoints.py	2	8	13
lingvo/jax/layers/normalizations.py	lingvo/jax/py_utils.py	2	8	12
lingvo/jax/layers/normalizations.py	lingvo/jax/model.py	2	8	24
lingvo/jax/layers/normalizations.py	lingvo/jax/eval.py	2	8	32
lingvo/jax/layers/normalizations.py	lingvo/jax/model_utils.py	2	8	9
lingvo/jax/layers/normalizations.py	lingvo/jax/train.py	2	8	24
lingvo/jax/layers/recurrent.py	lingvo/jax/checkpoints.py	2	6	13
lingvo/jax/layers/recurrent.py	lingvo/jax/layers/__init__.py	2	6	21
lingvo/jax/layers/recurrent.py	lingvo/jax/py_utils.py	2	6	12
lingvo/jax/layers/recurrent.py	lingvo/jax/model.py	2	6	24
lingvo/jax/layers/recurrent.py	lingvo/jax/layers/attentions.py	2	6	21
lingvo/jax/layers/recurrent.py	lingvo/jax/layers/embedding_softmax.py	2	6	11
lingvo/jax/layers/recurrent.py	lingvo/jax/optimizers.py	2	6	32
lingvo/jax/layers/recurrent.py	lingvo/jax/eval.py	2	6	32
lingvo/jax/layers/recurrent.py	lingvo/jax/model_utils.py	2	6	9
lingvo/jax/layers/recurrent.py	lingvo/jax/train.py	2	6	24
lingvo/jax/layers/recurrent.py	lingvo/jax/layers/convolutions.py	2	6	8
lingvo/jax/layers/recurrent.py	lingvo/jax/layers/resnets.py	2	6	4
lingvo/core/py_utils.py	lingvo/core/datasource.py	2	451	55
lingvo/core/py_utils.py	lingvo/core/saver.py	2	451	11
lingvo/jax/main.py	lingvo/jax/layers/pipeline.py	2	14	5
lingvo/jax/main.py	lingvo/jax/layers/__init__.py	2	14	21
lingvo/jax/main.py	lingvo/jax/layers/transformers.py	2	14	47
lingvo/jax/main.py	lingvo/jax/layers/attentions.py	2	14	21
lingvo/jax/main.py	lingvo/jax/layers/embedding_softmax.py	2	14	11
lingvo/jax/main.py	lingvo/jax/optimizers.py	2	14	32
lingvo/jax/main.py	lingvo/jax/layers/convolutions.py	2	14	8
lingvo/jax/main.py	lingvo/jax/layers/resnets.py	2	14	4
lingvo/jax/main.py	lingvo/jax/layers/normalizations.py	2	14	8
lingvo/jax/main.py	lingvo/jax/layers/recurrent.py	2	14	6
lingvo/core/layers.py	lingvo/tasks/mt/decoder.py	2	286	120
lingvo/core/layers.py	lingvo/core/saver.py	2	286	11
lingvo/jax/summary_utils.py	lingvo/jax/layers/__init__.py	2	7	21
lingvo/jax/summary_utils.py	lingvo/jax/model.py	2	7	24
lingvo/jax/summary_utils.py	lingvo/jax/layers/attentions.py	2	7	21
lingvo/jax/summary_utils.py	lingvo/jax/layers/embedding_softmax.py	2	7	11
lingvo/jax/summary_utils.py	lingvo/jax/optimizers.py	2	7	32
lingvo/jax/summary_utils.py	lingvo/jax/layers/convolutions.py	2	7	8
lingvo/jax/summary_utils.py	lingvo/jax/layers/resnets.py	2	7	4
lingvo/jax/summary_utils.py	lingvo/jax/layers/normalizations.py	2	7	8
lingvo/jax/summary_utils.py	lingvo/jax/layers/recurrent.py	2	7	6
lingvo/jax/layers/linears.py	lingvo/jax/checkpoints.py	2	7	13
lingvo/jax/layers/linears.py	lingvo/jax/py_utils.py	2	7	12
lingvo/jax/layers/linears.py	lingvo/jax/model.py	2	7	24
lingvo/jax/layers/linears.py	lingvo/jax/eval.py	2	7	32
lingvo/jax/layers/linears.py	lingvo/jax/trainer_lib.py	2	7	12
lingvo/jax/layers/linears.py	lingvo/jax/model_utils.py	2	7	9
lingvo/jax/layers/linears.py	lingvo/jax/train.py	2	7	24
lingvo/jax/layers/linears.py	lingvo/jax/base_layer.py	2	7	10
lingvo/jax/layers/linears.py	lingvo/jax/layers/recurrent.py	2	7	6
lingvo/jax/layers/linears.py	lingvo/jax/main.py	2	7	14
lingvo/jax/layers/linears.py	lingvo/jax/summary_utils.py	2	7	7
lingvo/core/gshard_layers.py	lingvo/core/saver.py	2	52	11
lingvo/core/gshard_layers.py	lingvo/core/layers.py	2	52	286
lingvo/core/base_input_generator.py	lingvo/core/py_utils.py	2	176	451
lingvo/jax/layers/repeats.py	lingvo/jax/checkpoints.py	2	5	13
lingvo/jax/layers/repeats.py	lingvo/jax/py_utils.py	2	5	12
lingvo/jax/layers/repeats.py	lingvo/jax/model.py	2	5	24
lingvo/jax/layers/repeats.py	lingvo/jax/eval.py	2	5	32
lingvo/jax/layers/repeats.py	lingvo/jax/trainer_lib.py	2	5	12
lingvo/jax/layers/repeats.py	lingvo/jax/model_utils.py	2	5	9
lingvo/jax/layers/repeats.py	lingvo/jax/train.py	2	5	24
lingvo/jax/layers/repeats.py	lingvo/jax/main.py	2	5	14
lingvo/base_runner.py	lingvo/core/program.py	2	84	136
lingvo/base_runner.py	lingvo/core/py_utils.py	2	84	451
lingvo/base_runner.py	lingvo/core/base_input_generator.py	2	84	176
lingvo/core/base_layer.py	lingvo/tasks/mt/decoder.py	2	132	120
lingvo/core/base_layer.py	lingvo/core/optimizer.py	2	132	56
lingvo/core/base_layer.py	lingvo/core/saver.py	2	132	11
lingvo/core/cluster.py	lingvo/executor.py	2	57	90
lingvo/core/cluster.py	lingvo/base_runner.py	2	57	84
lingvo/core/cluster.py	lingvo/core/base_layer.py	2	57	132
lingvo/core/cluster.py	lingvo/core/base_input_generator.py	2	57	176
lingvo/jax/checkpoint_managers.py	lingvo/jax/train.py	2	3	24
lingvo/eager_runners.py	lingvo/core/program.py	2	7	136
lingvo/core/schedule.py	lingvo/core/base_layer.py	2	40	132
lingvo/core/multitask_model.py	lingvo/core/base_model.py	2	11	213
lingvo/core/multitask_model.py	lingvo/core/gshard_layers.py	2	11	52
lingvo/core/multitask_model.py	lingvo/core/base_layer.py	2	11	132
lingvo/core/rnn_cell.py	lingvo/core/saver.py	2	118	11
lingvo/core/rnn_cell.py	lingvo/core/gshard_layers.py	2	118	52
lingvo/core/tokenizers.py	lingvo/core/base_input_generator.py	2	34	176
lingvo/core/builder_layers.py	lingvo/core/quant_utils.py	2	52	57
lingvo/core/builder_layers.py	lingvo/core/batch_major_attention.py	2	52	212
lingvo/core/conformer_layer.py	lingvo/core/builder_layers.py	2	55	52
lingvo/core/repeat_layer.py	lingvo/core/builder_layers.py	2	4	52
lingvo/core/repeat_layer.py	lingvo/core/conformer_layer.py	2	4	55
lingvo/core/repeat_layer.py	lingvo/core/batch_major_attention.py	2	4	212
lingvo/core/rnn_layers.py	lingvo/core/quant_utils.py	2	72	57
lingvo/core/rnn_layers.py	lingvo/core/saver.py	2	72	11
lingvo/core/attention.py	lingvo/tasks/mt/decoder.py	2	107	120
lingvo/core/attention.py	lingvo/core/batch_major_attention.py	2	107	212
lingvo/core/attention.py	lingvo/core/base_layer.py	2	107	132
lingvo/core/learner.py	lingvo/core/optimizer.py	2	52	56
lingvo/core/learner.py	lingvo/core/saver.py	2	52	11
lingvo/core/learner.py	lingvo/core/gshard_layers.py	2	52	52
lingvo/core/learner.py	lingvo/core/rnn_cell.py	2	52	118
lingvo/core/learner.py	lingvo/core/tpu_embedding_layers.py	2	52	42
lingvo/tasks/asr/fusion.py	lingvo/core/batch_major_attention.py	2	17	212
lingvo/jax/tasks/lm/input_generator.py	lingvo/jax/checkpoints.py	2	8	13
lingvo/jax/tasks/lm/input_generator.py	lingvo/jax/layers/__init__.py	2	8	21
lingvo/jax/tasks/lm/input_generator.py	lingvo/jax/layers/transformers.py	2	8	47
lingvo/jax/tasks/lm/input_generator.py	lingvo/jax/model.py	2	8	24
lingvo/jax/tasks/lm/input_generator.py	lingvo/jax/layers/attentions.py	2	8	21
lingvo/jax/tasks/lm/input_generator.py	lingvo/jax/layers/embedding_softmax.py	2	8	11
lingvo/jax/tasks/lm/input_generator.py	lingvo/jax/optimizers.py	2	8	32
lingvo/jax/tasks/lm/input_generator.py	lingvo/jax/trainer_lib.py	2	8	12
lingvo/jax/tasks/lm/input_generator.py	lingvo/jax/layers/convolutions.py	2	8	8
lingvo/jax/tasks/lm/input_generator.py	lingvo/jax/layers/resnets.py	2	8	4
lingvo/jax/tasks/lm/input_generator.py	lingvo/jax/base_layer.py	2	8	10
lingvo/jax/tasks/lm/input_generator.py	lingvo/jax/layers/normalizations.py	2	8	8
lingvo/jax/tasks/lm/input_generator.py	lingvo/jax/layers/recurrent.py	2	8	6
lingvo/jax/tasks/lm/input_generator.py	lingvo/jax/main.py	2	8	14
lingvo/jax/tasks/lm/input_generator.py	lingvo/jax/summary_utils.py	2	8	7
lingvo/jax/tasks/lm/input_generator.py	lingvo/jax/layers/linears.py	2	8	7
lingvo/jax/tasks/lm/input_generator.py	lingvo/jax/layers/repeats.py	2	8	5
lingvo/jax/gshard_utils.py	lingvo/jax/checkpoints.py	2	4	13
lingvo/jax/gshard_utils.py	lingvo/jax/layers/__init__.py	2	4	21
lingvo/jax/gshard_utils.py	lingvo/jax/py_utils.py	2	4	12
lingvo/jax/gshard_utils.py	lingvo/jax/model.py	2	4	24
lingvo/jax/gshard_utils.py	lingvo/jax/layers/attentions.py	2	4	21
lingvo/jax/gshard_utils.py	lingvo/jax/layers/embedding_softmax.py	2	4	11
lingvo/jax/gshard_utils.py	lingvo/jax/optimizers.py	2	4	32
lingvo/jax/gshard_utils.py	lingvo/jax/eval.py	2	4	32
lingvo/jax/gshard_utils.py	lingvo/jax/trainer_lib.py	2	4	12
lingvo/jax/gshard_utils.py	lingvo/jax/model_utils.py	2	4	9
lingvo/jax/gshard_utils.py	lingvo/jax/train.py	2	4	24
lingvo/jax/gshard_utils.py	lingvo/jax/layers/convolutions.py	2	4	8
lingvo/jax/gshard_utils.py	lingvo/jax/layers/resnets.py	2	4	4
lingvo/jax/gshard_utils.py	lingvo/jax/base_layer.py	2	4	10
lingvo/jax/gshard_utils.py	lingvo/jax/layers/normalizations.py	2	4	8
lingvo/jax/gshard_utils.py	lingvo/jax/layers/recurrent.py	2	4	6
lingvo/jax/gshard_utils.py	lingvo/jax/main.py	2	4	14
lingvo/jax/gshard_utils.py	lingvo/jax/summary_utils.py	2	4	7
lingvo/jax/gshard_utils.py	lingvo/jax/layers/linears.py	2	4	7
lingvo/jax/gshard_utils.py	lingvo/jax/layers/repeats.py	2	4	5
lingvo/jax/gshard_utils.py	lingvo/jax/tasks/lm/input_generator.py	2	4	8
lingvo/jax/tasks/lm/params/bert.py	lingvo/jax/checkpoints.py	2	8	13
lingvo/jax/tasks/lm/params/bert.py	lingvo/jax/py_utils.py	2	8	12
lingvo/jax/tasks/lm/params/bert.py	lingvo/jax/model.py	2	8	24
lingvo/jax/tasks/lm/params/bert.py	lingvo/jax/trainer_lib.py	2	8	12
lingvo/jax/tasks/lm/params/bert.py	lingvo/jax/model_utils.py	2	8	9
lingvo/jax/tasks/lm/params/bert.py	lingvo/jax/base_layer.py	2	8	10
lingvo/jax/tasks/lm/params/bert.py	lingvo/jax/layers/recurrent.py	2	8	6
lingvo/jax/tasks/lm/params/bert.py	lingvo/jax/main.py	2	8	14
lingvo/jax/tasks/lm/params/bert.py	lingvo/jax/summary_utils.py	2	8	7
lingvo/jax/tasks/lm/params/bert.py	lingvo/jax/gshard_utils.py	2	8	4
lingvo/trainer.py	lingvo/core/checkpointer.py	2	248	51
lingvo/tasks/car/geometry.py	lingvo/core/base_model.py	2	19	213
lingvo/tasks/car/geometry.py	lingvo/core/saver.py	2	19	11
lingvo/tasks/car/geometry.py	lingvo/core/batch_major_attention.py	2	19	212
lingvo/tasks/car/geometry.py	lingvo/core/py_utils.py	2	19	451
lingvo/tasks/car/geometry.py	lingvo/core/layers.py	2	19	286
lingvo/tasks/car/geometry.py	lingvo/core/gshard_layers.py	2	19	52
lingvo/tasks/car/geometry.py	lingvo/core/base_layer.py	2	19	132
lingvo/tasks/car/geometry.py	lingvo/core/rnn_cell.py	2	19	118
lingvo/tasks/car/geometry.py	lingvo/core/rnn_layers.py	2	19	72
lingvo/tasks/car/geometry.py	lingvo/core/learner.py	2	19	52
lingvo/trainer_utils.py	lingvo/executor.py	2	4	90
lingvo/trainer_utils.py	lingvo/eager_runners.py	2	4	7
lingvo/trainer_utils.py	lingvo/runners.py	2	4	6
lingvo/jax/base_model_params.py	lingvo/jax/checkpoints.py	2	4	13
lingvo/jax/base_model_params.py	lingvo/jax/base_input.py	2	4	14
lingvo/jax/base_model_params.py	lingvo/jax/layers/__init__.py	2	4	21
lingvo/jax/base_model_params.py	lingvo/jax/layers/transformers.py	2	4	47
lingvo/jax/base_model_params.py	lingvo/jax/py_utils.py	2	4	12
lingvo/jax/base_model_params.py	lingvo/jax/model.py	2	4	24
lingvo/jax/base_model_params.py	lingvo/jax/layers/attentions.py	2	4	21
lingvo/jax/base_model_params.py	lingvo/jax/layers/embedding_softmax.py	2	4	11
lingvo/jax/base_model_params.py	lingvo/jax/optimizers.py	2	4	32
lingvo/jax/base_model_params.py	lingvo/jax/trainer_lib.py	2	4	12
lingvo/jax/base_model_params.py	lingvo/jax/model_utils.py	2	4	9
lingvo/jax/base_model_params.py	lingvo/jax/layers/convolutions.py	2	4	8
lingvo/jax/base_model_params.py	lingvo/jax/layers/resnets.py	2	4	4
lingvo/jax/base_model_params.py	lingvo/jax/base_layer.py	2	4	10
lingvo/jax/base_model_params.py	lingvo/jax/layers/normalizations.py	2	4	8
lingvo/jax/base_model_params.py	lingvo/jax/layers/recurrent.py	2	4	6
lingvo/jax/base_model_params.py	lingvo/jax/main.py	2	4	14
lingvo/jax/base_model_params.py	lingvo/jax/summary_utils.py	2	4	7
lingvo/jax/base_model_params.py	lingvo/jax/layers/linears.py	2	4	7
lingvo/jax/base_model_params.py	lingvo/jax/layers/repeats.py	2	4	5
lingvo/jax/base_model_params.py	lingvo/jax/tasks/lm/input_generator.py	2	4	8
lingvo/jax/base_model_params.py	lingvo/jax/gshard_utils.py	2	4	4
lingvo/jax/schedules.py	lingvo/jax/checkpoints.py	2	4	13
lingvo/jax/schedules.py	lingvo/jax/py_utils.py	2	4	12
lingvo/jax/schedules.py	lingvo/jax/model.py	2	4	24
lingvo/jax/schedules.py	lingvo/jax/eval.py	2	4	32
lingvo/jax/schedules.py	lingvo/jax/trainer_lib.py	2	4	12
lingvo/jax/schedules.py	lingvo/jax/model_utils.py	2	4	9
lingvo/jax/schedules.py	lingvo/jax/train.py	2	4	24
lingvo/jax/schedules.py	lingvo/jax/base_layer.py	2	4	10
lingvo/jax/schedules.py	lingvo/jax/layers/recurrent.py	2	4	6
lingvo/jax/schedules.py	lingvo/jax/main.py	2	4	14
lingvo/jax/schedules.py	lingvo/jax/summary_utils.py	2	4	7
lingvo/jax/schedules.py	lingvo/jax/tasks/lm/input_generator.py	2	4	8
lingvo/jax/schedules.py	lingvo/jax/gshard_utils.py	2	4	4
lingvo/jax/schedules.py	lingvo/jax/base_model_params.py	2	4	4
lingvo/core/attention_util.py	lingvo/core/batch_major_attention.py	2	14	212
lingvo/core/ops/record_yielder.cc	lingvo/core/ops/record_batcher.cc	2	38	31
lingvo/lingvo.bzl	pip_package/build.Dockerfile	2	22	14
pip_package/setup.py	lingvo/lingvo.bzl	2	28	22
lingvo/jax/layers/activations.py	lingvo/jax/checkpoints.py	2	4	13
lingvo/jax/layers/activations.py	lingvo/jax/py_utils.py	2	4	12
lingvo/jax/layers/activations.py	lingvo/jax/model.py	2	4	24
lingvo/jax/layers/activations.py	lingvo/jax/eval.py	2	4	32
lingvo/jax/layers/activations.py	lingvo/jax/trainer_lib.py	2	4	12
lingvo/jax/layers/activations.py	lingvo/jax/model_utils.py	2	4	9
lingvo/jax/layers/activations.py	lingvo/jax/train.py	2	4	24
lingvo/jax/layers/activations.py	lingvo/jax/base_layer.py	2	4	10
lingvo/jax/layers/activations.py	lingvo/jax/layers/recurrent.py	2	4	6
lingvo/jax/layers/activations.py	lingvo/jax/main.py	2	4	14
lingvo/jax/layers/activations.py	lingvo/jax/summary_utils.py	2	4	7
lingvo/jax/layers/activations.py	lingvo/jax/tasks/lm/input_generator.py	2	4	8
lingvo/jax/layers/activations.py	lingvo/jax/gshard_utils.py	2	4	4
lingvo/jax/layers/activations.py	lingvo/jax/base_model_params.py	2	4	4
lingvo/core/summary_utils.py	lingvo/base_runner.py	2	49	84
lingvo/compat.py	lingvo/core/py_utils.py	2	70	451
lingvo/core/plot.py	lingvo/core/base_model.py	2	30	213
lingvo/core/plot.py	lingvo/core/saver.py	2	30	11
lingvo/core/plot.py	lingvo/core/batch_major_attention.py	2	30	212
lingvo/core/plot.py	lingvo/core/py_utils.py	2	30	451
lingvo/core/plot.py	lingvo/core/layers.py	2	30	286
lingvo/core/plot.py	lingvo/core/gshard_layers.py	2	30	52
lingvo/core/plot.py	lingvo/core/base_layer.py	2	30	132
lingvo/core/plot.py	lingvo/core/rnn_cell.py	2	30	118
lingvo/core/plot.py	lingvo/core/rnn_layers.py	2	30	72
lingvo/core/plot.py	lingvo/core/learner.py	2	30	52
lingvo/core/plot.py	lingvo/tasks/car/geometry.py	2	30	19
lingvo/jax/tasks/lm/params/params.py	lingvo/jax/tasks/lm/input_generator.py	2	2	8
lingvo/core/py_utils_flags.py	lingvo/core/py_utils.py	2	6	451
lingvo/jax/layers/poolings.py	lingvo/jax/checkpoints.py	2	3	13
lingvo/jax/layers/poolings.py	lingvo/jax/py_utils.py	2	3	12
lingvo/jax/layers/poolings.py	lingvo/jax/model.py	2	3	24
lingvo/jax/layers/poolings.py	lingvo/jax/eval.py	2	3	32
lingvo/jax/layers/poolings.py	lingvo/jax/trainer_lib.py	2	3	12
lingvo/jax/layers/poolings.py	lingvo/jax/model_utils.py	2	3	9
lingvo/jax/layers/poolings.py	lingvo/jax/train.py	2	3	24
lingvo/jax/layers/poolings.py	lingvo/jax/base_layer.py	2	3	10
lingvo/jax/layers/poolings.py	lingvo/jax/layers/recurrent.py	2	3	6
lingvo/jax/layers/poolings.py	lingvo/jax/main.py	2	3	14
lingvo/jax/layers/poolings.py	lingvo/jax/summary_utils.py	2	3	7
lingvo/jax/layers/poolings.py	lingvo/jax/tasks/lm/input_generator.py	2	3	8
lingvo/jax/layers/poolings.py	lingvo/jax/gshard_utils.py	2	3	4
lingvo/jax/layers/poolings.py	lingvo/jax/base_model_params.py	2	3	4
lingvo/jax/layers/stochastics.py	lingvo/jax/checkpoints.py	2	3	13
lingvo/jax/layers/stochastics.py	lingvo/jax/py_utils.py	2	3	12
lingvo/jax/layers/stochastics.py	lingvo/jax/model.py	2	3	24
lingvo/jax/layers/stochastics.py	lingvo/jax/eval.py	2	3	32
lingvo/jax/layers/stochastics.py	lingvo/jax/trainer_lib.py	2	3	12
lingvo/jax/layers/stochastics.py	lingvo/jax/model_utils.py	2	3	9
lingvo/jax/layers/stochastics.py	lingvo/jax/train.py	2	3	24
lingvo/jax/layers/stochastics.py	lingvo/jax/base_layer.py	2	3	10
lingvo/jax/layers/stochastics.py	lingvo/jax/layers/recurrent.py	2	3	6
lingvo/jax/layers/stochastics.py	lingvo/jax/main.py	2	3	14
lingvo/jax/layers/stochastics.py	lingvo/jax/summary_utils.py	2	3	7
lingvo/jax/layers/stochastics.py	lingvo/jax/tasks/lm/input_generator.py	2	3	8
lingvo/jax/layers/stochastics.py	lingvo/jax/gshard_utils.py	2	3	4
lingvo/jax/layers/stochastics.py	lingvo/jax/base_model_params.py	2	3	4
lingvo/jax/layers/augmentations.py	lingvo/jax/checkpoints.py	2	2	13
lingvo/jax/layers/augmentations.py	lingvo/jax/layers/__init__.py	2	2	21
lingvo/jax/layers/augmentations.py	lingvo/jax/layers/transformers.py	2	2	47
lingvo/jax/layers/augmentations.py	lingvo/jax/py_utils.py	2	2	12
lingvo/jax/layers/augmentations.py	lingvo/jax/model.py	2	2	24
lingvo/jax/layers/augmentations.py	lingvo/jax/layers/attentions.py	2	2	21
lingvo/jax/layers/augmentations.py	lingvo/jax/layers/embedding_softmax.py	2	2	11
lingvo/jax/layers/augmentations.py	lingvo/jax/optimizers.py	2	2	32
lingvo/jax/layers/augmentations.py	lingvo/jax/eval.py	2	2	32
lingvo/jax/layers/augmentations.py	lingvo/jax/trainer_lib.py	2	2	12
lingvo/jax/layers/augmentations.py	lingvo/jax/model_utils.py	2	2	9
lingvo/jax/layers/augmentations.py	lingvo/jax/train.py	2	2	24
lingvo/jax/layers/augmentations.py	lingvo/jax/layers/convolutions.py	2	2	8
lingvo/jax/layers/augmentations.py	lingvo/jax/layers/resnets.py	2	2	4
lingvo/jax/layers/augmentations.py	lingvo/jax/base_layer.py	2	2	10
lingvo/jax/layers/augmentations.py	lingvo/jax/layers/normalizations.py	2	2	8
lingvo/jax/layers/augmentations.py	lingvo/jax/layers/recurrent.py	2	2	6
lingvo/jax/layers/augmentations.py	lingvo/jax/main.py	2	2	14
lingvo/jax/layers/augmentations.py	lingvo/jax/summary_utils.py	2	2	7
lingvo/jax/layers/augmentations.py	lingvo/jax/layers/linears.py	2	2	7
lingvo/jax/layers/augmentations.py	lingvo/jax/layers/repeats.py	2	2	5
lingvo/jax/layers/augmentations.py	lingvo/jax/tasks/lm/input_generator.py	2	2	8
lingvo/jax/layers/augmentations.py	lingvo/jax/gshard_utils.py	2	2	4
lingvo/jax/layers/augmentations.py	lingvo/jax/tasks/lm/params/bert.py	2	2	8
lingvo/jax/layers/augmentations.py	lingvo/jax/base_model_params.py	2	2	4
lingvo/jax/layers/augmentations.py	lingvo/jax/schedules.py	2	2	4
lingvo/jax/layers/augmentations.py	lingvo/jax/layers/activations.py	2	2	4
lingvo/jax/layers/augmentations.py	lingvo/jax/layers/poolings.py	2	2	3
lingvo/jax/layers/augmentations.py	lingvo/jax/layers/stochastics.py	2	2	3
lingvo/jax/model_imports.py	lingvo/jax/checkpoints.py	2	2	13
lingvo/jax/model_imports.py	lingvo/jax/layers/__init__.py	2	2	21
lingvo/jax/model_imports.py	lingvo/jax/layers/transformers.py	2	2	47
lingvo/jax/model_imports.py	lingvo/jax/py_utils.py	2	2	12
lingvo/jax/model_imports.py	lingvo/jax/model.py	2	2	24
lingvo/jax/model_imports.py	lingvo/jax/layers/attentions.py	2	2	21
lingvo/jax/model_imports.py	lingvo/jax/layers/embedding_softmax.py	2	2	11
lingvo/jax/model_imports.py	lingvo/jax/optimizers.py	2	2	32
lingvo/jax/model_imports.py	lingvo/jax/eval.py	2	2	32
lingvo/jax/model_imports.py	lingvo/jax/trainer_lib.py	2	2	12
lingvo/jax/model_imports.py	lingvo/jax/model_utils.py	2	2	9
lingvo/jax/model_imports.py	lingvo/jax/train.py	2	2	24
lingvo/jax/model_imports.py	lingvo/jax/layers/convolutions.py	2	2	8
lingvo/jax/model_imports.py	lingvo/jax/layers/resnets.py	2	2	4
lingvo/jax/model_imports.py	lingvo/jax/base_layer.py	2	2	10
lingvo/jax/model_imports.py	lingvo/jax/layers/normalizations.py	2	2	8
lingvo/jax/model_imports.py	lingvo/jax/layers/recurrent.py	2	2	6
lingvo/jax/model_imports.py	lingvo/jax/main.py	2	2	14
lingvo/jax/model_imports.py	lingvo/jax/summary_utils.py	2	2	7
lingvo/jax/model_imports.py	lingvo/jax/layers/linears.py	2	2	7
lingvo/jax/model_imports.py	lingvo/jax/layers/repeats.py	2	2	5
lingvo/jax/model_imports.py	lingvo/jax/tasks/lm/input_generator.py	2	2	8
lingvo/jax/model_imports.py	lingvo/jax/gshard_utils.py	2	2	4
lingvo/jax/model_imports.py	lingvo/jax/tasks/lm/params/bert.py	2	2	8
lingvo/jax/model_imports.py	lingvo/jax/base_model_params.py	2	2	4
lingvo/jax/model_imports.py	lingvo/jax/schedules.py	2	2	4
lingvo/jax/model_imports.py	lingvo/jax/layers/activations.py	2	2	4
lingvo/jax/model_imports.py	lingvo/jax/layers/poolings.py	2	2	3
lingvo/jax/model_imports.py	lingvo/jax/layers/stochastics.py	2	2	3
lingvo/jax/model_imports.py	lingvo/jax/layers/augmentations.py	2	2	2
lingvo/jax/model_registry.py	lingvo/jax/checkpoints.py	2	2	13
lingvo/jax/model_registry.py	lingvo/jax/layers/__init__.py	2	2	21
lingvo/jax/model_registry.py	lingvo/jax/layers/transformers.py	2	2	47
lingvo/jax/model_registry.py	lingvo/jax/py_utils.py	2	2	12
lingvo/jax/model_registry.py	lingvo/jax/model.py	2	2	24
lingvo/jax/model_registry.py	lingvo/jax/layers/attentions.py	2	2	21
lingvo/jax/model_registry.py	lingvo/jax/layers/embedding_softmax.py	2	2	11
lingvo/jax/model_registry.py	lingvo/jax/optimizers.py	2	2	32
lingvo/jax/model_registry.py	lingvo/jax/eval.py	2	2	32
lingvo/jax/model_registry.py	lingvo/jax/trainer_lib.py	2	2	12
lingvo/jax/model_registry.py	lingvo/jax/model_utils.py	2	2	9
lingvo/jax/model_registry.py	lingvo/jax/train.py	2	2	24
lingvo/jax/model_registry.py	lingvo/jax/layers/convolutions.py	2	2	8
lingvo/jax/model_registry.py	lingvo/jax/layers/resnets.py	2	2	4
lingvo/jax/model_registry.py	lingvo/jax/base_layer.py	2	2	10
lingvo/jax/model_registry.py	lingvo/jax/layers/normalizations.py	2	2	8
lingvo/jax/model_registry.py	lingvo/jax/layers/recurrent.py	2	2	6
lingvo/jax/model_registry.py	lingvo/jax/main.py	2	2	14
lingvo/jax/model_registry.py	lingvo/jax/summary_utils.py	2	2	7
lingvo/jax/model_registry.py	lingvo/jax/layers/linears.py	2	2	7
lingvo/jax/model_registry.py	lingvo/jax/layers/repeats.py	2	2	5
lingvo/jax/model_registry.py	lingvo/jax/tasks/lm/input_generator.py	2	2	8
lingvo/jax/model_registry.py	lingvo/jax/gshard_utils.py	2	2	4
lingvo/jax/model_registry.py	lingvo/jax/tasks/lm/params/bert.py	2	2	8
lingvo/jax/model_registry.py	lingvo/jax/base_model_params.py	2	2	4
lingvo/jax/model_registry.py	lingvo/jax/schedules.py	2	2	4
lingvo/jax/model_registry.py	lingvo/jax/layers/activations.py	2	2	4
lingvo/jax/model_registry.py	lingvo/jax/layers/poolings.py	2	2	3
lingvo/jax/model_registry.py	lingvo/jax/layers/stochastics.py	2	2	3
lingvo/jax/model_registry.py	lingvo/jax/layers/augmentations.py	2	2	2
lingvo/jax/model_registry.py	lingvo/jax/model_imports.py	2	2	2
lingvo/jax/train_states.py	lingvo/jax/checkpoints.py	2	2	13
lingvo/jax/train_states.py	lingvo/jax/layers/__init__.py	2	2	21
lingvo/jax/train_states.py	lingvo/jax/layers/transformers.py	2	2	47
lingvo/jax/train_states.py	lingvo/jax/py_utils.py	2	2	12
lingvo/jax/train_states.py	lingvo/jax/model.py	2	2	24
lingvo/jax/train_states.py	lingvo/jax/layers/attentions.py	2	2	21
lingvo/jax/train_states.py	lingvo/jax/layers/embedding_softmax.py	2	2	11
lingvo/jax/train_states.py	lingvo/jax/optimizers.py	2	2	32
lingvo/jax/train_states.py	lingvo/jax/eval.py	2	2	32
lingvo/jax/train_states.py	lingvo/jax/trainer_lib.py	2	2	12
lingvo/jax/train_states.py	lingvo/jax/model_utils.py	2	2	9
lingvo/jax/train_states.py	lingvo/jax/train.py	2	2	24
lingvo/jax/train_states.py	lingvo/jax/layers/convolutions.py	2	2	8
lingvo/jax/train_states.py	lingvo/jax/layers/resnets.py	2	2	4
lingvo/jax/train_states.py	lingvo/jax/base_layer.py	2	2	10
lingvo/jax/train_states.py	lingvo/jax/layers/normalizations.py	2	2	8
lingvo/jax/train_states.py	lingvo/jax/layers/recurrent.py	2	2	6
lingvo/jax/train_states.py	lingvo/jax/main.py	2	2	14
lingvo/jax/train_states.py	lingvo/jax/summary_utils.py	2	2	7
lingvo/jax/train_states.py	lingvo/jax/layers/linears.py	2	2	7
lingvo/jax/train_states.py	lingvo/jax/layers/repeats.py	2	2	5
lingvo/jax/train_states.py	lingvo/jax/tasks/lm/input_generator.py	2	2	8
lingvo/jax/train_states.py	lingvo/jax/gshard_utils.py	2	2	4
lingvo/jax/train_states.py	lingvo/jax/tasks/lm/params/bert.py	2	2	8
lingvo/jax/train_states.py	lingvo/jax/base_model_params.py	2	2	4
lingvo/jax/train_states.py	lingvo/jax/schedules.py	2	2	4
lingvo/jax/train_states.py	lingvo/jax/layers/activations.py	2	2	4
lingvo/jax/train_states.py	lingvo/jax/layers/poolings.py	2	2	3
lingvo/jax/train_states.py	lingvo/jax/layers/stochastics.py	2	2	3
lingvo/jax/train_states.py	lingvo/jax/layers/augmentations.py	2	2	2
lingvo/jax/train_states.py	lingvo/jax/model_imports.py	2	2	2
lingvo/jax/train_states.py	lingvo/jax/model_registry.py	2	2	2
lingvo/core/recurrent.py	lingvo/core/py_utils.py	2	68	451
lingvo/core/recurrent.py	lingvo/core/gshard_layers.py	2	68	52
lingvo/core/layers_with_attention.py	lingvo/core/py_utils.py	2	87	451
lingvo/core/layers_with_attention.py	lingvo/core/layers.py	2	87	286
lingvo/core/layers_with_attention.py	lingvo/tasks/lm/model.py	2	87	36
lingvo/core/beam_search_helper.py	lingvo/core/ops/beam_search_step_op_kernels.h	2	71	16
lingvo/core/ml_perf_log.py	lingvo/tasks/lm/params/wiki_bert.py	2	5	8
lingvo/tasks/lm/tools/gshard_lm_decode.py	lingvo/core/gshard_decode.py	2	2	5
lingvo/core/gshard_utils.py	lingvo/core/gshard_layers.py	2	12	52
lingvo/core/gshard_utils.py	lingvo/core/py_utils.py	2	12	451
lingvo/tasks/asr/model.py	lingvo/core/gshard_layers.py	2	50	52
lingvo/core/conv_layers_with_time_padding.py	lingvo/core/quant_utils.py	2	46	57
lingvo/datasets.py	lingvo/core/base_model_params.py	2	11	15
lingvo/tasks/lm/tokenizer.py	lingvo/core/layers.py	2	2	286
lingvo/tasks/lm/tokenizer.py	lingvo/core/gshard_builder.py	2	2	86
lingvo/tasks/lm/tokenizer.py	lingvo/tasks/lm/input_generator.py	2	2	24
lingvo/tasks/asr/decoder.py	lingvo/core/base_model.py	2	89	213
lingvo/tasks/asr/decoder.py	lingvo/core/saver.py	2	89	11
lingvo/tasks/asr/decoder.py	lingvo/core/batch_major_attention.py	2	89	212
lingvo/tasks/asr/decoder.py	lingvo/core/py_utils.py	2	89	451
lingvo/tasks/asr/decoder.py	lingvo/core/layers.py	2	89	286
lingvo/tasks/asr/decoder.py	lingvo/core/gshard_layers.py	2	89	52
lingvo/tasks/asr/decoder.py	lingvo/core/base_layer.py	2	89	132
lingvo/tasks/asr/decoder.py	lingvo/core/rnn_cell.py	2	89	118
lingvo/tasks/asr/decoder.py	lingvo/core/rnn_layers.py	2	89	72
lingvo/tasks/asr/decoder.py	lingvo/core/learner.py	2	89	52
lingvo/tasks/asr/decoder.py	lingvo/tasks/car/geometry.py	2	89	19
lingvo/tasks/asr/decoder.py	lingvo/core/plot.py	2	89	30
lingvo/core/hyperparams.py	lingvo/core/base_model.py	2	64	213
lingvo/core/hyperparams.py	lingvo/core/saver.py	2	64	11
lingvo/core/hyperparams.py	lingvo/core/layers.py	2	64	286
lingvo/core/hyperparams.py	lingvo/core/gshard_layers.py	2	64	52
lingvo/core/hyperparams.py	lingvo/core/rnn_cell.py	2	64	118
lingvo/core/hyperparams.py	lingvo/core/rnn_layers.py	2	64	72
lingvo/core/hyperparams.py	lingvo/core/learner.py	2	64	52
lingvo/core/hyperparams.py	lingvo/tasks/car/geometry.py	2	64	19
lingvo/core/hyperparams.py	lingvo/core/plot.py	2	64	30
lingvo/core/hyperparams.py	lingvo/tasks/asr/decoder.py	2	64	89
lingvo/core/var_tmp_wrappers.py	lingvo/core/gshard_layers.py	2	2	52
lingvo/core/cluster_factory.py	lingvo/core/cluster.py	2	22	57
lingvo/tasks/car/ap_metric.py	lingvo/core/base_model.py	2	9	213
lingvo/tasks/car/ap_metric.py	lingvo/core/saver.py	2	9	11
lingvo/tasks/car/ap_metric.py	lingvo/core/batch_major_attention.py	2	9	212
lingvo/tasks/car/ap_metric.py	lingvo/core/py_utils.py	2	9	451
lingvo/tasks/car/ap_metric.py	lingvo/core/layers.py	2	9	286
lingvo/tasks/car/ap_metric.py	lingvo/core/gshard_layers.py	2	9	52
lingvo/tasks/car/ap_metric.py	lingvo/core/base_layer.py	2	9	132
lingvo/tasks/car/ap_metric.py	lingvo/core/rnn_cell.py	2	9	118
lingvo/tasks/car/ap_metric.py	lingvo/core/rnn_layers.py	2	9	72
lingvo/tasks/car/ap_metric.py	lingvo/core/learner.py	2	9	52
lingvo/tasks/car/ap_metric.py	lingvo/tasks/car/geometry.py	2	9	19
lingvo/tasks/car/ap_metric.py	lingvo/core/plot.py	2	9	30
lingvo/tasks/car/ap_metric.py	lingvo/tasks/asr/decoder.py	2	9	89
lingvo/tasks/car/ap_metric.py	lingvo/core/hyperparams.py	2	9	64
lingvo/tasks/car/calibration_processing.py	lingvo/core/base_model.py	2	8	213
lingvo/tasks/car/calibration_processing.py	lingvo/core/saver.py	2	8	11
lingvo/tasks/car/calibration_processing.py	lingvo/core/batch_major_attention.py	2	8	212
lingvo/tasks/car/calibration_processing.py	lingvo/core/py_utils.py	2	8	451
lingvo/tasks/car/calibration_processing.py	lingvo/core/layers.py	2	8	286
lingvo/tasks/car/calibration_processing.py	lingvo/core/gshard_layers.py	2	8	52
lingvo/tasks/car/calibration_processing.py	lingvo/core/base_layer.py	2	8	132
lingvo/tasks/car/calibration_processing.py	lingvo/core/rnn_cell.py	2	8	118
lingvo/tasks/car/calibration_processing.py	lingvo/core/rnn_layers.py	2	8	72
lingvo/tasks/car/calibration_processing.py	lingvo/core/learner.py	2	8	52
lingvo/tasks/car/calibration_processing.py	lingvo/tasks/car/geometry.py	2	8	19
lingvo/tasks/car/calibration_processing.py	lingvo/core/plot.py	2	8	30
lingvo/tasks/car/calibration_processing.py	lingvo/tasks/asr/decoder.py	2	8	89
lingvo/tasks/car/calibration_processing.py	lingvo/core/hyperparams.py	2	8	64
lingvo/tasks/car/calibration_processing.py	lingvo/tasks/car/ap_metric.py	2	8	9
lingvo/tasks/car/kitti_ap_metric.py	lingvo/core/base_model.py	2	7	213
lingvo/tasks/car/kitti_ap_metric.py	lingvo/core/saver.py	2	7	11
lingvo/tasks/car/kitti_ap_metric.py	lingvo/core/batch_major_attention.py	2	7	212
lingvo/tasks/car/kitti_ap_metric.py	lingvo/core/py_utils.py	2	7	451
lingvo/tasks/car/kitti_ap_metric.py	lingvo/core/layers.py	2	7	286
lingvo/tasks/car/kitti_ap_metric.py	lingvo/core/gshard_layers.py	2	7	52
lingvo/tasks/car/kitti_ap_metric.py	lingvo/core/base_layer.py	2	7	132
lingvo/tasks/car/kitti_ap_metric.py	lingvo/core/rnn_cell.py	2	7	118
lingvo/tasks/car/kitti_ap_metric.py	lingvo/core/rnn_layers.py	2	7	72
lingvo/tasks/car/kitti_ap_metric.py	lingvo/core/learner.py	2	7	52
lingvo/tasks/car/kitti_ap_metric.py	lingvo/tasks/car/geometry.py	2	7	19
lingvo/tasks/car/kitti_ap_metric.py	lingvo/core/plot.py	2	7	30
lingvo/tasks/car/kitti_ap_metric.py	lingvo/tasks/asr/decoder.py	2	7	89
lingvo/tasks/car/kitti_ap_metric.py	lingvo/core/hyperparams.py	2	7	64
lingvo/tasks/car/kitti_ap_metric.py	lingvo/tasks/car/ap_metric.py	2	7	9
lingvo/tasks/car/kitti_ap_metric.py	lingvo/tasks/car/calibration_processing.py	2	7	8
lingvo/tasks/car/pointnet.py	lingvo/core/base_model.py	2	5	213
lingvo/tasks/car/pointnet.py	lingvo/core/saver.py	2	5	11
lingvo/tasks/car/pointnet.py	lingvo/core/batch_major_attention.py	2	5	212
lingvo/tasks/car/pointnet.py	lingvo/core/py_utils.py	2	5	451
lingvo/tasks/car/pointnet.py	lingvo/core/layers.py	2	5	286
lingvo/tasks/car/pointnet.py	lingvo/core/gshard_layers.py	2	5	52
lingvo/tasks/car/pointnet.py	lingvo/core/base_layer.py	2	5	132
lingvo/tasks/car/pointnet.py	lingvo/core/rnn_cell.py	2	5	118
lingvo/tasks/car/pointnet.py	lingvo/core/rnn_layers.py	2	5	72
lingvo/tasks/car/pointnet.py	lingvo/core/learner.py	2	5	52
lingvo/tasks/car/pointnet.py	lingvo/tasks/car/geometry.py	2	5	19
lingvo/tasks/car/pointnet.py	lingvo/core/plot.py	2	5	30
lingvo/tasks/car/pointnet.py	lingvo/tasks/asr/decoder.py	2	5	89
lingvo/tasks/car/pointnet.py	lingvo/core/hyperparams.py	2	5	64
lingvo/tasks/car/pointnet.py	lingvo/tasks/car/ap_metric.py	2	5	9
lingvo/tasks/car/pointnet.py	lingvo/tasks/car/calibration_processing.py	2	5	8
lingvo/tasks/car/pointnet.py	lingvo/tasks/car/kitti_ap_metric.py	2	5	7
lingvo/tasks/lm/layers.py	lingvo/core/rnn_layers.py	2	43	72
lingvo/tasks/lm/layers.py	lingvo/core/layers.py	2	43	286
lingvo/tasks/lm/layers.py	lingvo/tasks/lm/model.py	2	43	36
lingvo/core/thread_local_utils.py	lingvo/core/py_utils.py	2	3	451
lingvo/tasks/milan/utils.py	lingvo/tasks/milan/dataset_spec.py	2	2	4
lingvo/core/ops/__init__.py	lingvo/core/ops/beam_search_step_op_kernels.cc	2	15	46
lingvo/model_imports.py	pip_package/build.Dockerfile	2	22	14
lingvo/model_imports.py	pip_package/setup.py	2	22	28
lingvo/core/pruning_utils.py	lingvo/core/layers.py	2	5	286
lingvo/core/saver.py	lingvo/executor.py	1	11	90
lingvo/jax/base_input.py	lingvo/jax/checkpoints.py	1	14	13
lingvo/jax/layers/__init__.py	lingvo/jax/base_input.py	1	21	14
lingvo/jax/py_utils.py	lingvo/jax/layers/pipeline.py	1	12	5
lingvo/core/datasource.py	lingvo/tasks/punctuator/input_generator.py	1	55	24
lingvo/core/datasource.py	lingvo/core/base_model.py	1	55	213
lingvo/core/datasource.py	lingvo/executor.py	1	55	90
lingvo/tasks/punctuator/model.py	lingvo/tasks/punctuator/input_generator.py	1	12	24
lingvo/tasks/punctuator/model.py	lingvo/core/datasource.py	1	12	55
lingvo/tasks/punctuator/params/codelab.py	lingvo/tasks/punctuator/input_generator.py	1	15	24
lingvo/tasks/punctuator/params/codelab.py	lingvo/core/datasource.py	1	15	55
lingvo/tasks/punctuator/params/codelab.py	lingvo/tasks/punctuator/model.py	1	15	12
lingvo/core/metrics.py	lingvo/executor.py	1	38	90
lingvo/tasks/mt/decoder.py	lingvo/core/base_decoder.py	1	120	33
lingvo/jax/layers/attentions.py	lingvo/jax/layers/conformers.py	1	21	5
lingvo/jax/layers/attentions.py	lingvo/jax/base_input.py	1	21	14
lingvo/jax/layers/embedding_softmax.py	lingvo/jax/base_input.py	1	11	14
lingvo/jax/optimizers.py	lingvo/jax/base_input.py	1	32	14
lingvo/core/optimizer.py	lingvo/executor.py	1	56	90
lingvo/core/optimizer.py	lingvo/core/program.py	1	56	136
lingvo/model_registry.py	lingvo/core/base_model.py	1	31	213
lingvo/jax/trainer_lib.py	lingvo/jax/learners.py	1	12	8
lingvo/jax/trainer_lib.py	lingvo/jax/base_input.py	1	12	14
lingvo/jax/tasks/lm/model_params.py	lingvo/jax/model.py	1	8	24
lingvo/core/batch_major_attention.py	lingvo/jax/eval.py	1	212	32
lingvo/core/batch_major_attention.py	lingvo/jax/train.py	1	212	24
lingvo/jax/layers/spectrum_augmenter.py	lingvo/jax/layers/__init__.py	1	1	21
lingvo/jax/layers/convolutions.py	lingvo/jax/base_input.py	1	8	14
lingvo/jax/layers/resnets.py	lingvo/jax/base_input.py	1	4	14
lingvo/tasks/mt/input_generator.py	lingvo/core/scorers.py	1	41	6
lingvo/tasks/mt/model.py	lingvo/core/scorers.py	1	43	6
lingvo/tasks/mt/model.py	lingvo/tasks/mt/input_generator.py	1	43	41
lingvo/tasks/mt/params/params.py	lingvo/core/scorers.py	1	4	6
lingvo/tasks/mt/params/params.py	lingvo/tasks/mt/input_generator.py	1	4	41
lingvo/tasks/mt/params/params.py	lingvo/tasks/mt/model.py	1	4	43
lingvo/tasks/mt/params/xendec/wmt14_en_de.py	lingvo/core/scorers.py	1	1	6
lingvo/tasks/mt/params/xendec/wmt14_en_de.py	lingvo/tasks/mt/input_generator.py	1	1	41
lingvo/tasks/mt/params/xendec/wmt14_en_de.py	lingvo/tasks/mt/model.py	1	1	43
lingvo/tasks/mt/params/xendec/wmt14_en_de.py	lingvo/tasks/mt/params/params.py	1	1	4
lingvo/jax/base_layer.py	lingvo/jax/base_input.py	1	10	14
lingvo/jax/layers/ngrammer.py	lingvo/jax/trainer_lib.py	1	8	12
lingvo/jax/layers/ngrammer.py	lingvo/jax/base_layer.py	1	8	10
lingvo/jax/layers/ngrammer.py	lingvo/jax/layers/__init__.py	1	8	21
lingvo/jax/layers/normalizations.py	lingvo/jax/layers/ngrammer.py	1	8	8
lingvo/jax/layers/normalizations.py	lingvo/jax/base_input.py	1	8	14
lingvo/jax/layers/recurrent.py	lingvo/jax/layers/ngrammer.py	1	6	8
lingvo/jax/layers/recurrent.py	lingvo/jax/layers/pipeline.py	1	6	5
lingvo/jax/layers/recurrent.py	lingvo/jax/base_input.py	1	6	14
lingvo/core/py_utils.py	lingvo/jax/layers/convolutions.py	1	451	8
lingvo/core/py_utils.py	lingvo/tasks/mt/decoder.py	1	451	120
lingvo/jax/layers/ctc_objectives.py	lingvo/jax/layers/__init__.py	1	2	21
lingvo/jax/main.py	lingvo/jax/base_input.py	1	14	14
lingvo/core/layers.py	lingvo/core/datasource.py	1	286	55
lingvo/core/layers.py	lingvo/jax/layers/normalizations.py	1	286	8
lingvo/jax/summary_utils.py	lingvo/jax/tasks/lm/model_params.py	1	7	8
lingvo/jax/summary_utils.py	lingvo/jax/base_input.py	1	7	14
lingvo/models_test_helper.py	lingvo/core/optimizer.py	1	37	56
lingvo/models_test_helper.py	lingvo/core/py_utils.py	1	37	451
lingvo/models_test_helper.py	lingvo/core/datasource.py	1	37	55
lingvo/jax/layers/linears.py	lingvo/jax/base_input.py	1	7	14
lingvo/core/gshard_layers.py	lingvo/executor.py	1	52	90
lingvo/core/gshard_layers.py	lingvo/tasks/mt/decoder.py	1	52	120
lingvo/core/gshard_layers.py	lingvo/jax/layers/transformers.py	1	52	47
lingvo/jax/asserts.py	lingvo/jax/layers/transformers.py	1	3	47
lingvo/core/inference_graph_exporter.py	lingvo/core/base_model.py	1	55	213
lingvo/core/predictor.py	lingvo/core/inference_graph_exporter.py	1	39	55
lingvo/core/predictor.py	lingvo/core/py_utils.py	1	39	451
lingvo/core/base_input_generator.py	lingvo/core/checkpointer.py	1	176	51
lingvo/core/base_input_generator.py	lingvo/models_test_helper.py	1	176	37
lingvo/core/tpu_embedding_layers.py	lingvo/core/inference_graph_exporter.py	1	42	55
lingvo/core/tpu_embedding_layers.py	lingvo/core/program.py	1	42	136
lingvo/core/tpu_embedding_layers.py	lingvo/tasks/mt/model.py	1	42	43
lingvo/core/tpu_embedding_layers.py	lingvo/executor.py	1	42	90
lingvo/jax/layers/repeats.py	lingvo/jax/tasks/lm/model_params.py	1	5	8
lingvo/jax/layers/repeats.py	lingvo/jax/layers/pipeline.py	1	5	5
lingvo/jax/layers/repeats.py	lingvo/jax/base_input.py	1	5	14
lingvo/core/quant_utils.py	lingvo/core/base_model.py	1	57	213
lingvo/core/quant_utils.py	lingvo/tasks/mt/decoder.py	1	57	120
lingvo/core/quant_utils.py	lingvo/core/gshard_layers.py	1	57	52
lingvo/core/quant_utils.py	lingvo/core/tpu_embedding_layers.py	1	57	42
lingvo/core/quant_utils.py	lingvo/core/batch_major_attention.py	1	57	212
lingvo/jax/tasks/lm/params/lm_cloud.py	lingvo/jax/layers/transformers.py	1	7	47
lingvo/jax/tasks/lm/params/lm_cloud.py	lingvo/jax/layers/recurrent.py	1	7	6
lingvo/jax/tasks/lm/params/lm_cloud.py	lingvo/jax/train.py	1	7	24
lingvo/jax/tasks/lm/params/lm_cloud.py	lingvo/jax/main.py	1	7	14
lingvo/base_runner.py	lingvo/core/checkpointer.py	1	84	51
lingvo/base_runner.py	lingvo/core/datasource.py	1	84	55
lingvo/core/base_layer.py	lingvo/base_runner.py	1	132	84
lingvo/core/base_layer.py	lingvo/core/datasource.py	1	132	55
lingvo/core/base_layer.py	lingvo/core/quant_utils.py	1	132	57
lingvo/core/base_layer.py	lingvo/core/inference_graph_exporter.py	1	132	55
lingvo/core/cluster.py	lingvo/core/checkpointer.py	1	57	51
lingvo/core/cluster.py	lingvo/models_test_helper.py	1	57	37
lingvo/jax/checkpoint_managers.py	lingvo/jax/checkpoints.py	1	3	13
lingvo/jax/checkpoint_managers.py	lingvo/jax/model.py	1	3	24
lingvo/eager_runners.py	lingvo/core/base_model.py	1	7	213
lingvo/eager_runners.py	lingvo/core/base_layer.py	1	7	132
lingvo/eager_runners.py	lingvo/core/checkpointer.py	1	7	51
lingvo/runners.py	lingvo/core/base_model.py	1	6	213
lingvo/runners.py	lingvo/core/program.py	1	6	136
lingvo/runners.py	lingvo/core/base_layer.py	1	6	132
lingvo/core/schedule.py	lingvo/core/datasource.py	1	40	55
lingvo/core/schedule.py	lingvo/core/layers.py	1	40	286
lingvo/core/schedule.py	lingvo/core/gshard_builder.py	1	40	86
lingvo/core/schedule.py	lingvo/core/optimizer.py	1	40	56
lingvo/core/multitask_model.py	lingvo/executor.py	1	11	90
lingvo/core/multitask_model.py	lingvo/tasks/mt/decoder.py	1	11	120
lingvo/core/multitask_model.py	lingvo/core/quant_utils.py	1	11	57
lingvo/core/multitask_model.py	lingvo/core/py_utils.py	1	11	451
lingvo/core/spectrum_augmenter.py	lingvo/core/datasource.py	1	28	55
lingvo/core/spectrum_augmenter.py	lingvo/core/layers.py	1	28	286
lingvo/core/spectrum_augmenter.py	lingvo/core/base_layer.py	1	28	132
lingvo/core/spectrum_augmenter.py	lingvo/core/schedule.py	1	28	40
lingvo/core/task_scheduler.py	lingvo/core/datasource.py	1	18	55
lingvo/core/task_scheduler.py	lingvo/core/layers.py	1	18	286
lingvo/core/task_scheduler.py	lingvo/core/base_layer.py	1	18	132
lingvo/core/task_scheduler.py	lingvo/core/schedule.py	1	18	40
lingvo/core/task_scheduler.py	lingvo/core/spectrum_augmenter.py	1	18	28
lingvo/core/task_scheduler.py	lingvo/core/program.py	1	18	136
lingvo/core/tokenizers.py	lingvo/core/datasource.py	1	34	55
lingvo/core/tokenizers.py	lingvo/core/layers.py	1	34	286
lingvo/core/tokenizers.py	lingvo/core/base_layer.py	1	34	132
lingvo/core/tokenizers.py	lingvo/core/schedule.py	1	34	40
lingvo/core/tokenizers.py	lingvo/core/spectrum_augmenter.py	1	34	28
lingvo/core/tokenizers.py	lingvo/core/task_scheduler.py	1	34	18
lingvo/tasks/car/input_extractor.py	lingvo/core/datasource.py	1	13	55
lingvo/tasks/car/input_extractor.py	lingvo/core/layers.py	1	13	286
lingvo/tasks/car/input_extractor.py	lingvo/core/base_layer.py	1	13	132
lingvo/tasks/car/input_extractor.py	lingvo/core/schedule.py	1	13	40
lingvo/tasks/car/input_extractor.py	lingvo/core/spectrum_augmenter.py	1	13	28
lingvo/tasks/car/input_extractor.py	lingvo/core/task_scheduler.py	1	13	18
lingvo/tasks/car/input_extractor.py	lingvo/core/tokenizers.py	1	13	34
lingvo/tasks/mt/data_augmenter.py	lingvo/core/datasource.py	1	5	55
lingvo/tasks/mt/data_augmenter.py	lingvo/core/layers.py	1	5	286
lingvo/tasks/mt/data_augmenter.py	lingvo/core/base_layer.py	1	5	132
lingvo/tasks/mt/data_augmenter.py	lingvo/core/schedule.py	1	5	40
lingvo/tasks/mt/data_augmenter.py	lingvo/core/spectrum_augmenter.py	1	5	28
lingvo/tasks/mt/data_augmenter.py	lingvo/core/task_scheduler.py	1	5	18
lingvo/tasks/mt/data_augmenter.py	lingvo/core/tokenizers.py	1	5	34
lingvo/tasks/mt/data_augmenter.py	lingvo/tasks/car/input_extractor.py	1	5	13
lingvo/tasks/milan/tf_hub_layers.py	lingvo/core/base_layer.py	1	2	132
lingvo/core/builder_layers.py	lingvo/core/base_model.py	1	52	213
lingvo/core/builder_layers.py	lingvo/tasks/mt/decoder.py	1	52	120
lingvo/core/builder_layers.py	lingvo/core/base_layer.py	1	52	132
lingvo/core/builder_layers.py	lingvo/core/multitask_model.py	1	52	11
lingvo/core/builder_layers.py	lingvo/core/layers.py	1	52	286
lingvo/core/builder_layers.py	lingvo/core/rnn_cell.py	1	52	118
lingvo/core/conformer_layer.py	lingvo/core/base_model.py	1	55	213
lingvo/core/conformer_layer.py	lingvo/tasks/mt/decoder.py	1	55	120
lingvo/core/conformer_layer.py	lingvo/core/quant_utils.py	1	55	57
lingvo/core/conformer_layer.py	lingvo/core/base_layer.py	1	55	132
lingvo/core/conformer_layer.py	lingvo/core/multitask_model.py	1	55	11
lingvo/core/conformer_layer.py	lingvo/core/py_utils.py	1	55	451
lingvo/core/gpipe.py	lingvo/core/base_model.py	1	34	213
lingvo/core/gpipe.py	lingvo/tasks/mt/decoder.py	1	34	120
lingvo/core/gpipe.py	lingvo/core/gshard_layers.py	1	34	52
lingvo/core/gpipe.py	lingvo/core/quant_utils.py	1	34	57
lingvo/core/gpipe.py	lingvo/core/base_layer.py	1	34	132
lingvo/core/gpipe.py	lingvo/core/multitask_model.py	1	34	11
lingvo/core/gpipe.py	lingvo/core/builder_layers.py	1	34	52
lingvo/core/gpipe.py	lingvo/core/conformer_layer.py	1	34	55
lingvo/core/repeat_layer.py	lingvo/core/base_model.py	1	4	213
lingvo/core/repeat_layer.py	lingvo/tasks/mt/decoder.py	1	4	120
lingvo/core/repeat_layer.py	lingvo/core/gshard_layers.py	1	4	52
lingvo/core/repeat_layer.py	lingvo/core/quant_utils.py	1	4	57
lingvo/core/repeat_layer.py	lingvo/core/base_layer.py	1	4	132
lingvo/core/repeat_layer.py	lingvo/core/multitask_model.py	1	4	11
lingvo/core/repeat_layer.py	lingvo/core/gpipe.py	1	4	34
lingvo/core/rnn_layers.py	lingvo/core/multitask_model.py	1	72	11
lingvo/core/rnn_layers.py	lingvo/core/builder_layers.py	1	72	52
lingvo/core/rnn_layers.py	lingvo/core/conformer_layer.py	1	72	55
lingvo/core/rnn_layers.py	lingvo/core/gpipe.py	1	72	34
lingvo/core/rnn_layers.py	lingvo/core/repeat_layer.py	1	72	4
lingvo/jax/layers/losses.py	lingvo/jax/layers/__init__.py	1	1	21
lingvo/jax/layers/losses.py	lingvo/jax/layers/embedding_softmax.py	1	1	11
lingvo/jax/pip_package/build.Dockerfile	lingvo/jax/layers/__init__.py	1	3	21
lingvo/jax/pip_package/build.Dockerfile	lingvo/jax/layers/embedding_softmax.py	1	3	11
lingvo/jax/pip_package/build.Dockerfile	lingvo/jax/layers/losses.py	1	3	1
lingvo/jax/pip_package/build.Dockerfile	lingvo/jax/optimizers.py	1	3	32
lingvo/jax/pip_package/setup.py	lingvo/jax/layers/__init__.py	1	4	21
lingvo/jax/pip_package/setup.py	lingvo/jax/layers/embedding_softmax.py	1	4	11
lingvo/jax/pip_package/setup.py	lingvo/jax/layers/losses.py	1	4	1
lingvo/jax/pip_package/setup.py	lingvo/jax/optimizers.py	1	4	32
lingvo/jax/checkpoint.proto	lingvo/jax/checkpoints.py	1	1	13
lingvo/jax/checkpoint.proto	lingvo/jax/model.py	1	1	24
lingvo/jax/checkpoint.proto	lingvo/jax/train.py	1	1	24
lingvo/jax/checkpoint.proto	lingvo/jax/checkpoint_managers.py	1	1	3
lingvo/jax/optimizer_prefix_vectorization.py	lingvo/jax/learners.py	1	2	8
lingvo/jax/optimizer_prefix_vectorization.py	lingvo/jax/model.py	1	2	24
lingvo/jax/optimizer_prefix_vectorization.py	lingvo/jax/optimizers.py	1	2	32
lingvo/jax/optimizer_prefix_vectorization.py	lingvo/jax/trainer_lib.py	1	2	12
lingvo/tasks/milan/dual_encoder.py	lingvo/core/tpu_embedding_layers.py	1	2	42
lingvo/tasks/milan/dual_encoder.py	lingvo/core/quant_utils.py	1	2	57
lingvo/tasks/milan/dual_encoder.py	lingvo/tasks/milan/tf_hub_layers.py	1	2	2
lingvo/tasks/mt/encoder.py	lingvo/core/tpu_embedding_layers.py	1	48	42
lingvo/tasks/mt/encoder.py	lingvo/core/quant_utils.py	1	48	57
lingvo/tasks/mt/encoder.py	lingvo/tasks/milan/dual_encoder.py	1	48	2
lingvo/jax/layers/vanillanets.py	lingvo/jax/layers/__init__.py	1	2	21
lingvo/jax/layers/vanillanets.py	lingvo/jax/layers/convolutions.py	1	2	8
lingvo/jax/layers/vanillanets.py	lingvo/core/py_utils.py	1	2	451
lingvo/core/attention.py	lingvo/core/base_model.py	1	107	213
lingvo/core/attention.py	lingvo/core/builder_layers.py	1	107	52
lingvo/core/layers_with_gpipe.py	lingvo/core/base_model.py	1	42	213
lingvo/core/layers_with_gpipe.py	lingvo/tasks/mt/decoder.py	1	42	120
lingvo/core/layers_with_gpipe.py	lingvo/core/batch_major_attention.py	1	42	212
lingvo/core/layers_with_gpipe.py	lingvo/core/layers.py	1	42	286
lingvo/core/layers_with_gpipe.py	lingvo/core/base_layer.py	1	42	132
lingvo/core/layers_with_gpipe.py	lingvo/core/rnn_layers.py	1	42	72
lingvo/core/layers_with_gpipe.py	lingvo/core/attention.py	1	42	107
lingvo/core/learner.py	lingvo/tasks/mt/decoder.py	1	52	120
lingvo/core/learner.py	lingvo/core/attention.py	1	52	107
lingvo/core/learner.py	lingvo/core/layers_with_gpipe.py	1	52	42
lingvo/core/learner.py	lingvo/core/program.py	1	52	136
lingvo/core/learner.py	lingvo/core/base_input_generator.py	1	52	176
lingvo/core/learner.py	lingvo/core/schedule.py	1	52	40
lingvo/core/step.py	lingvo/core/base_model.py	1	19	213
lingvo/core/step.py	lingvo/tasks/mt/decoder.py	1	19	120
lingvo/core/step.py	lingvo/core/batch_major_attention.py	1	19	212
lingvo/core/step.py	lingvo/core/layers.py	1	19	286
lingvo/core/step.py	lingvo/core/base_layer.py	1	19	132
lingvo/core/step.py	lingvo/core/rnn_layers.py	1	19	72
lingvo/core/step.py	lingvo/core/attention.py	1	19	107
lingvo/core/step.py	lingvo/core/layers_with_gpipe.py	1	19	42
lingvo/core/step.py	lingvo/core/learner.py	1	19	52
lingvo/core/steps/rnn_steps.py	lingvo/core/base_model.py	1	13	213
lingvo/core/steps/rnn_steps.py	lingvo/tasks/mt/decoder.py	1	13	120
lingvo/core/steps/rnn_steps.py	lingvo/core/batch_major_attention.py	1	13	212
lingvo/core/steps/rnn_steps.py	lingvo/core/layers.py	1	13	286
lingvo/core/steps/rnn_steps.py	lingvo/core/base_layer.py	1	13	132
lingvo/core/steps/rnn_steps.py	lingvo/core/rnn_layers.py	1	13	72
lingvo/core/steps/rnn_steps.py	lingvo/core/attention.py	1	13	107
lingvo/core/steps/rnn_steps.py	lingvo/core/layers_with_gpipe.py	1	13	42
lingvo/core/steps/rnn_steps.py	lingvo/core/learner.py	1	13	52
lingvo/core/steps/rnn_steps.py	lingvo/core/step.py	1	13	19
lingvo/tasks/asr/fusion.py	lingvo/core/base_model.py	1	17	213
lingvo/tasks/asr/fusion.py	lingvo/tasks/mt/decoder.py	1	17	120
lingvo/tasks/asr/fusion.py	lingvo/core/layers.py	1	17	286
lingvo/tasks/asr/fusion.py	lingvo/core/base_layer.py	1	17	132
lingvo/tasks/asr/fusion.py	lingvo/core/rnn_layers.py	1	17	72
lingvo/tasks/asr/fusion.py	lingvo/core/attention.py	1	17	107
lingvo/tasks/asr/fusion.py	lingvo/core/layers_with_gpipe.py	1	17	42
lingvo/tasks/asr/fusion.py	lingvo/core/learner.py	1	17	52
lingvo/tasks/asr/fusion.py	lingvo/core/step.py	1	17	19
lingvo/tasks/asr/fusion.py	lingvo/core/steps/rnn_steps.py	1	17	13
lingvo/core/activations.py	lingvo/core/layers.py	1	6	286
lingvo/core/ops/input_common.cc	lingvo/core/datasource.py	1	12	55
lingvo/core/ops/input_common.cc	lingvo/core/base_input_generator.py	1	12	176
lingvo/core/trainer_test_utils.py	lingvo/executor.py	1	12	90
lingvo/core/trainer_test_utils.py	lingvo/core/base_model.py	1	12	213
lingvo/tasks/asr/decoder_metrics.py	lingvo/core/base_decoder.py	1	14	33
lingvo/jax/layers/stats.py	lingvo/jax/layers/pipeline.py	1	1	5
lingvo/jax/layers/stats.py	lingvo/jax/layers/transformers.py	1	1	47
lingvo/jax/layers/stats.py	lingvo/jax/base_layer.py	1	1	10
lingvo/jax/layers/stats.py	lingvo/jax/layers/recurrent.py	1	1	6
lingvo/jax/layers/stats.py	lingvo/jax/layers/repeats.py	1	1	5
lingvo/jax/tasks/lm/input_generator.py	lingvo/jax/tasks/lm/params/lm_cloud.py	1	8	7
lingvo/jax/gshard_utils.py	lingvo/core/gshard_layers.py	1	4	52
lingvo/jax/gshard_utils.py	lingvo/jax/base_input.py	1	4	14
lingvo/core/builder.py	lingvo/core/batch_major_attention.py	1	10	212
lingvo/core/builder.py	lingvo/core/gshard_builder.py	1	10	86
lingvo/core/builder.py	lingvo/core/conformer_layer.py	1	10	55
lingvo/jax/tasks/lm/params/bert.py	lingvo/jax/tasks/lm/model_params.py	1	8	8
lingvo/jax/tasks/lm/params/bert.py	lingvo/jax/tasks/lm/params/lm_cloud.py	1	8	7
lingvo/trainer.py	lingvo/core/predictor.py	1	248	39
lingvo/trainer.py	lingvo/core/inference_graph_exporter.py	1	248	55
lingvo/trainer.py	lingvo/core/base_layer.py	1	248	132
lingvo/trainer.py	lingvo/core/tpu_embedding_layers.py	1	248	42
lingvo/tasks/lm/model.py	lingvo/core/py_utils.py	1	36	451
lingvo/tasks/lm/model.py	lingvo/core/gshard_layers.py	1	36	52
lingvo/tasks/lm/model.py	lingvo/core/batch_major_attention.py	1	36	212
lingvo/tasks/lm/model.py	lingvo/core/layers.py	1	36	286
lingvo/tasks/lm/model.py	lingvo/core/rnn_layers.py	1	36	72
lingvo/jax/io_utils.py	lingvo/jax/model.py	1	1	24
lingvo/jax/io_utils.py	lingvo/jax/eval.py	1	1	32
lingvo/trainer_utils.py	lingvo/core/checkpointer.py	1	4	51
lingvo/trainer_utils.py	lingvo/core/py_utils.py	1	4	451
lingvo/trainer_utils.py	lingvo/core/cluster.py	1	4	57
lingvo/jax/schedules.py	lingvo/jax/base_input.py	1	4	14
lingvo/jax/pip_package/MANIFEST.in	lingvo/jax/pip_package/build.Dockerfile	1	1	3
lingvo/jax/pip_package/MANIFEST.in	lingvo/jax/pip_package/setup.py	1	1	4
lingvo/core/attention_util.py	lingvo/core/conformer_layer.py	1	14	55
lingvo/core/attention_util.py	lingvo/core/py_utils.py	1	14	451
lingvo/core/target_sequence_sampler.py	lingvo/tasks/mt/decoder.py	1	19	120
pip_package/build.Dockerfile	lingvo/tasks/milan/tf_hub_layers.py	1	14	2
pip_package/build.Dockerfile	lingvo/tasks/milan/dual_encoder.py	1	14	2
pip_package/build.Dockerfile	lingvo/core/datasource.py	1	14	55
lingvo/core/ops/beam_search_step_op_kernels.cc	lingvo/core/ops/input_common.cc	1	46	12
lingvo/core/ops/best_step_op_kernels.cc	lingvo/core/ops/input_common.cc	1	7	12
lingvo/core/ops/best_step_op_kernels.cc	lingvo/core/ops/beam_search_step_op_kernels.cc	1	7	46
lingvo/core/ops/generic_input_op_kernels.cc	lingvo/core/ops/input_common.cc	1	20	12
lingvo/core/ops/generic_input_op_kernels.cc	lingvo/core/ops/beam_search_step_op_kernels.cc	1	20	46
lingvo/core/ops/generic_input_op_kernels.cc	lingvo/core/ops/best_step_op_kernels.cc	1	20	7
lingvo/core/ops/input_common.h	lingvo/core/ops/beam_search_step_op_kernels.cc	1	16	46
lingvo/core/ops/input_common.h	lingvo/core/ops/best_step_op_kernels.cc	1	16	7
lingvo/core/ops/input_common.h	lingvo/core/ops/generic_input_op_kernels.cc	1	16	20
lingvo/core/ops/input_common.h	lingvo/core/datasource.py	1	16	55
lingvo/core/ops/input_common.h	lingvo/core/base_input_generator.py	1	16	176
lingvo/core/ops/pack_ops.cc	lingvo/core/ops/input_common.cc	1	9	12
lingvo/core/ops/pack_ops.cc	lingvo/core/ops/beam_search_step_op_kernels.cc	1	9	46
lingvo/core/ops/pack_ops.cc	lingvo/core/ops/best_step_op_kernels.cc	1	9	7
lingvo/core/ops/pack_ops.cc	lingvo/core/ops/generic_input_op_kernels.cc	1	9	20
lingvo/core/ops/pack_ops.cc	lingvo/core/ops/input_common.h	1	9	16
lingvo/core/ops/pack_ops.cc	lingvo/core/ops/x_ops.cc	1	9	55
lingvo/core/ops/random_ops_kernels.cc	lingvo/core/ops/input_common.cc	1	3	12
lingvo/core/ops/random_ops_kernels.cc	lingvo/core/ops/beam_search_step_op_kernels.cc	1	3	46
lingvo/core/ops/random_ops_kernels.cc	lingvo/core/ops/best_step_op_kernels.cc	1	3	7
lingvo/core/ops/random_ops_kernels.cc	lingvo/core/ops/generic_input_op_kernels.cc	1	3	20
lingvo/core/ops/random_ops_kernels.cc	lingvo/core/ops/input_common.h	1	3	16
lingvo/core/ops/random_ops_kernels.cc	lingvo/core/ops/pack_ops.cc	1	3	9
lingvo/core/ops/record_batcher.cc	lingvo/core/ops/input_common.cc	1	31	12
lingvo/core/ops/record_batcher.cc	lingvo/core/ops/beam_search_step_op_kernels.cc	1	31	46
lingvo/core/ops/record_batcher.cc	lingvo/core/ops/best_step_op_kernels.cc	1	31	7
lingvo/core/ops/record_batcher.cc	lingvo/core/ops/generic_input_op_kernels.cc	1	31	20
lingvo/core/ops/record_batcher.cc	lingvo/core/ops/input_common.h	1	31	16
lingvo/core/ops/record_batcher.cc	lingvo/core/ops/pack_ops.cc	1	31	9
lingvo/core/ops/record_batcher.cc	lingvo/core/ops/random_ops_kernels.cc	1	31	3
lingvo/core/ops/record_batcher.h	lingvo/core/ops/input_common.cc	1	16	12
lingvo/core/ops/record_batcher.h	lingvo/core/ops/beam_search_step_op_kernels.cc	1	16	46
lingvo/core/ops/record_batcher.h	lingvo/core/ops/best_step_op_kernels.cc	1	16	7
lingvo/core/ops/record_batcher.h	lingvo/core/ops/generic_input_op_kernels.cc	1	16	20
lingvo/core/ops/record_batcher.h	lingvo/core/ops/input_common.h	1	16	16
lingvo/core/ops/record_batcher.h	lingvo/core/ops/pack_ops.cc	1	16	9
lingvo/core/ops/record_batcher.h	lingvo/core/ops/random_ops_kernels.cc	1	16	3
lingvo/core/ops/record_batcher.h	lingvo/core/ops/record_batcher.cc	1	16	31
lingvo/core/ops/record_yielder.cc	lingvo/core/ops/input_common.cc	1	38	12
lingvo/core/ops/record_yielder.cc	lingvo/core/ops/beam_search_step_op_kernels.cc	1	38	46
lingvo/core/ops/record_yielder.cc	lingvo/core/ops/best_step_op_kernels.cc	1	38	7
lingvo/core/ops/record_yielder.cc	lingvo/core/ops/generic_input_op_kernels.cc	1	38	20
lingvo/core/ops/record_yielder.cc	lingvo/core/ops/input_common.h	1	38	16
lingvo/core/ops/record_yielder.cc	lingvo/core/ops/pack_ops.cc	1	38	9
lingvo/core/ops/record_yielder.cc	lingvo/core/ops/random_ops_kernels.cc	1	38	3
lingvo/core/ops/record_yielder.cc	lingvo/core/ops/record_batcher.h	1	38	16
lingvo/core/ops/record_yielder.cc	lingvo/core/datasource.py	1	38	55
lingvo/core/ops/record_yielder.h	lingvo/core/ops/input_common.cc	1	20	12
lingvo/core/ops/record_yielder.h	lingvo/core/ops/beam_search_step_op_kernels.cc	1	20	46
lingvo/core/ops/record_yielder.h	lingvo/core/ops/best_step_op_kernels.cc	1	20	7
lingvo/core/ops/record_yielder.h	lingvo/core/ops/generic_input_op_kernels.cc	1	20	20
lingvo/core/ops/record_yielder.h	lingvo/core/ops/input_common.h	1	20	16
lingvo/core/ops/record_yielder.h	lingvo/core/ops/pack_ops.cc	1	20	9
lingvo/core/ops/record_yielder.h	lingvo/core/ops/random_ops_kernels.cc	1	20	3
lingvo/core/ops/record_yielder.h	lingvo/core/ops/record_batcher.cc	1	20	31
lingvo/core/ops/record_yielder.h	lingvo/core/ops/record_batcher.h	1	20	16
lingvo/core/ops/record_yielder.h	lingvo/core/ops/record_yielder.cc	1	20	38
lingvo/core/ops/static_map_op.cc	lingvo/core/ops/input_common.cc	1	4	12
lingvo/core/ops/static_map_op.cc	lingvo/core/ops/beam_search_step_op_kernels.cc	1	4	46
lingvo/core/ops/static_map_op.cc	lingvo/core/ops/best_step_op_kernels.cc	1	4	7
lingvo/core/ops/static_map_op.cc	lingvo/core/ops/generic_input_op_kernels.cc	1	4	20
lingvo/core/ops/static_map_op.cc	lingvo/core/ops/input_common.h	1	4	16
lingvo/core/ops/static_map_op.cc	lingvo/core/ops/pack_ops.cc	1	4	9
lingvo/core/ops/static_map_op.cc	lingvo/core/ops/random_ops_kernels.cc	1	4	3
lingvo/core/ops/static_map_op.cc	lingvo/core/ops/record_batcher.cc	1	4	31
lingvo/core/ops/static_map_op.cc	lingvo/core/ops/record_batcher.h	1	4	16
lingvo/core/ops/static_map_op.cc	lingvo/core/ops/record_yielder.cc	1	4	38
lingvo/core/ops/static_map_op.cc	lingvo/core/ops/record_yielder.h	1	4	20
lingvo/core/ops/weighted_mix_record_yielder.cc	lingvo/core/ops/input_common.cc	1	8	12
lingvo/core/ops/weighted_mix_record_yielder.cc	lingvo/core/ops/beam_search_step_op_kernels.cc	1	8	46
lingvo/core/ops/weighted_mix_record_yielder.cc	lingvo/core/ops/best_step_op_kernels.cc	1	8	7
lingvo/core/ops/weighted_mix_record_yielder.cc	lingvo/core/ops/generic_input_op_kernels.cc	1	8	20
lingvo/core/ops/weighted_mix_record_yielder.cc	lingvo/core/ops/input_common.h	1	8	16
lingvo/core/ops/weighted_mix_record_yielder.cc	lingvo/core/ops/pack_ops.cc	1	8	9
lingvo/core/ops/weighted_mix_record_yielder.cc	lingvo/core/ops/random_ops_kernels.cc	1	8	3
lingvo/core/ops/weighted_mix_record_yielder.cc	lingvo/core/ops/record_batcher.cc	1	8	31
lingvo/core/ops/weighted_mix_record_yielder.cc	lingvo/core/ops/record_batcher.h	1	8	16
lingvo/core/ops/weighted_mix_record_yielder.cc	lingvo/core/ops/record_yielder.cc	1	8	38
lingvo/core/ops/weighted_mix_record_yielder.cc	lingvo/core/ops/record_yielder.h	1	8	20
lingvo/core/ops/weighted_mix_record_yielder.cc	lingvo/core/ops/static_map_op.cc	1	8	4
lingvo/core/ops/weighted_mix_record_yielder.h	lingvo/core/ops/input_common.cc	1	6	12
lingvo/core/ops/weighted_mix_record_yielder.h	lingvo/core/ops/beam_search_step_op_kernels.cc	1	6	46
lingvo/core/ops/weighted_mix_record_yielder.h	lingvo/core/ops/best_step_op_kernels.cc	1	6	7
lingvo/core/ops/weighted_mix_record_yielder.h	lingvo/core/ops/generic_input_op_kernels.cc	1	6	20
lingvo/core/ops/weighted_mix_record_yielder.h	lingvo/core/ops/input_common.h	1	6	16
lingvo/core/ops/weighted_mix_record_yielder.h	lingvo/core/ops/pack_ops.cc	1	6	9
lingvo/core/ops/weighted_mix_record_yielder.h	lingvo/core/ops/random_ops_kernels.cc	1	6	3
lingvo/core/ops/weighted_mix_record_yielder.h	lingvo/core/ops/record_batcher.cc	1	6	31
lingvo/core/ops/weighted_mix_record_yielder.h	lingvo/core/ops/record_batcher.h	1	6	16
lingvo/core/ops/weighted_mix_record_yielder.h	lingvo/core/ops/record_yielder.cc	1	6	38
lingvo/core/ops/weighted_mix_record_yielder.h	lingvo/core/ops/record_yielder.h	1	6	20
lingvo/core/ops/weighted_mix_record_yielder.h	lingvo/core/ops/static_map_op.cc	1	6	4
lingvo/core/ops/weighted_mix_record_yielder.h	lingvo/core/ops/weighted_mix_record_yielder.cc	1	6	8
lingvo/core/ops/yielder_test_helper.h	lingvo/core/ops/input_common.cc	1	5	12
lingvo/core/ops/yielder_test_helper.h	lingvo/core/ops/beam_search_step_op_kernels.cc	1	5	46
lingvo/core/ops/yielder_test_helper.h	lingvo/core/ops/best_step_op_kernels.cc	1	5	7
lingvo/core/ops/yielder_test_helper.h	lingvo/core/ops/generic_input_op_kernels.cc	1	5	20
lingvo/core/ops/yielder_test_helper.h	lingvo/core/ops/input_common.h	1	5	16
lingvo/core/ops/yielder_test_helper.h	lingvo/core/ops/pack_ops.cc	1	5	9
lingvo/core/ops/yielder_test_helper.h	lingvo/core/ops/random_ops_kernels.cc	1	5	3
lingvo/core/ops/yielder_test_helper.h	lingvo/core/ops/record_batcher.cc	1	5	31
lingvo/core/ops/yielder_test_helper.h	lingvo/core/ops/record_batcher.h	1	5	16
lingvo/core/ops/yielder_test_helper.h	lingvo/core/ops/record_yielder.cc	1	5	38
lingvo/core/ops/yielder_test_helper.h	lingvo/core/ops/record_yielder.h	1	5	20
lingvo/core/ops/yielder_test_helper.h	lingvo/core/ops/static_map_op.cc	1	5	4
lingvo/core/ops/yielder_test_helper.h	lingvo/core/ops/weighted_mix_record_yielder.cc	1	5	8
lingvo/core/ops/yielder_test_helper.h	lingvo/core/ops/weighted_mix_record_yielder.h	1	5	6
lingvo/tasks/car/ops/ps_utils.cc	lingvo/core/ops/input_common.cc	1	8	12
lingvo/tasks/car/ops/ps_utils.cc	lingvo/core/ops/beam_search_step_op_kernels.cc	1	8	46
lingvo/tasks/car/ops/ps_utils.cc	lingvo/core/ops/best_step_op_kernels.cc	1	8	7
lingvo/tasks/car/ops/ps_utils.cc	lingvo/core/ops/generic_input_op_kernels.cc	1	8	20
lingvo/tasks/car/ops/ps_utils.cc	lingvo/core/ops/input_common.h	1	8	16
lingvo/tasks/car/ops/ps_utils.cc	lingvo/core/ops/pack_ops.cc	1	8	9
lingvo/tasks/car/ops/ps_utils.cc	lingvo/core/ops/random_ops_kernels.cc	1	8	3
lingvo/tasks/car/ops/ps_utils.cc	lingvo/core/ops/record_batcher.cc	1	8	31
lingvo/tasks/car/ops/ps_utils.cc	lingvo/core/ops/record_batcher.h	1	8	16
lingvo/tasks/car/ops/ps_utils.cc	lingvo/core/ops/record_yielder.cc	1	8	38
lingvo/tasks/car/ops/ps_utils.cc	lingvo/core/ops/record_yielder.h	1	8	20
lingvo/tasks/car/ops/ps_utils.cc	lingvo/core/ops/static_map_op.cc	1	8	4
lingvo/tasks/car/ops/ps_utils.cc	lingvo/core/ops/weighted_mix_record_yielder.cc	1	8	8
lingvo/tasks/car/ops/ps_utils.cc	lingvo/core/ops/weighted_mix_record_yielder.h	1	8	6
lingvo/tasks/car/ops/ps_utils.cc	lingvo/core/ops/yielder_test_helper.h	1	8	5
lingvo/lingvo.bzl	lingvo/core/batch_major_attention.py	1	22	212
lingvo/lingvo.bzl	lingvo/core/py_utils.py	1	22	451
lingvo/lingvo.bzl	lingvo/core/base_layer.py	1	22	132
lingvo/lingvo.bzl	lingvo/tasks/milan/tf_hub_layers.py	1	22	2
lingvo/lingvo.bzl	lingvo/tasks/milan/dual_encoder.py	1	22	2
lingvo/repo.bzl	pip_package/build.Dockerfile	1	17	14
lingvo/repo.bzl	lingvo/lingvo.bzl	1	17	22
pip_package/setup.py	lingvo/repo.bzl	1	28	17
pip_package/setup.py	lingvo/tasks/milan/tf_hub_layers.py	1	28	2
pip_package/setup.py	lingvo/tasks/milan/dual_encoder.py	1	28	2
pip_package/setup.py	lingvo/core/datasource.py	1	28	55
lingvo/jax/layers/activations.py	lingvo/jax/base_input.py	1	4	14
lingvo/jax/pytypes.py	lingvo/jax/checkpoints.py	1	2	13
lingvo/jax/pytypes.py	lingvo/jax/layers/__init__.py	1	2	21
lingvo/jax/pytypes.py	lingvo/jax/layers/transformers.py	1	2	47
lingvo/jax/pytypes.py	lingvo/jax/py_utils.py	1	2	12
lingvo/jax/pytypes.py	lingvo/jax/model.py	1	2	24
lingvo/jax/pytypes.py	lingvo/jax/layers/attentions.py	1	2	21
lingvo/jax/pytypes.py	lingvo/jax/layers/embedding_softmax.py	1	2	11
lingvo/jax/pytypes.py	lingvo/jax/optimizers.py	1	2	32
lingvo/jax/pytypes.py	lingvo/jax/eval.py	1	2	32
lingvo/jax/pytypes.py	lingvo/jax/trainer_lib.py	1	2	12
lingvo/jax/pytypes.py	lingvo/jax/model_utils.py	1	2	9
lingvo/jax/pytypes.py	lingvo/jax/train.py	1	2	24
lingvo/jax/pytypes.py	lingvo/jax/layers/convolutions.py	1	2	8
lingvo/jax/pytypes.py	lingvo/jax/layers/resnets.py	1	2	4
lingvo/jax/pytypes.py	lingvo/jax/base_layer.py	1	2	10
lingvo/jax/pytypes.py	lingvo/jax/layers/normalizations.py	1	2	8
lingvo/jax/pytypes.py	lingvo/jax/layers/recurrent.py	1	2	6
lingvo/jax/pytypes.py	lingvo/jax/main.py	1	2	14
lingvo/jax/pytypes.py	lingvo/jax/summary_utils.py	1	2	7
lingvo/jax/pytypes.py	lingvo/jax/layers/linears.py	1	2	7
lingvo/jax/pytypes.py	lingvo/jax/layers/repeats.py	1	2	5
lingvo/jax/pytypes.py	lingvo/jax/tasks/lm/input_generator.py	1	2	8
lingvo/jax/pytypes.py	lingvo/jax/gshard_utils.py	1	2	4
lingvo/jax/pytypes.py	lingvo/jax/tasks/lm/params/bert.py	1	2	8
lingvo/jax/pytypes.py	lingvo/jax/base_model_params.py	1	2	4
lingvo/jax/pytypes.py	lingvo/jax/schedules.py	1	2	4
lingvo/jax/pytypes.py	lingvo/jax/layers/activations.py	1	2	4
lingvo/core/summary_utils.py	lingvo/core/cluster.py	1	49	57
lingvo/core/summary_utils.py	lingvo/core/program.py	1	49	136
lingvo/core/summary_utils.py	lingvo/core/base_input_generator.py	1	49	176
lingvo/core/nested_map.py	lingvo/core/program.py	1	8	136
lingvo/core/nested_map.py	lingvo/core/batch_major_attention.py	1	8	212
lingvo/core/nested_map.py	lingvo/core/py_utils.py	1	8	451
lingvo/core/nested_map.py	lingvo/core/base_layer.py	1	8	132
lingvo/core/nested_map.py	lingvo/lingvo.bzl	1	8	22
lingvo/compat.py	lingvo/core/inference_graph_exporter.py	1	70	55
lingvo/compat.py	lingvo/trainer.py	1	70	248
lingvo/compat.py	lingvo/core/summary_utils.py	1	70	49
lingvo/tasks/milan/dataset_spec.py	lingvo/core/py_utils.py	1	4	451
lingvo/tasks/milan/dataset_spec.py	lingvo/compat.py	1	4	70
lingvo/tasks/milan/dataset_spec.py	lingvo/tasks/milan/tf_hub_layers.py	1	4	2
lingvo/tasks/milan/dataset_spec.py	lingvo/tasks/milan/dual_encoder.py	1	4	2
lingvo/tasks/milan/dataset_spec.py	pip_package/build.Dockerfile	1	4	14
lingvo/tasks/milan/dataset_spec.py	lingvo/lingvo.bzl	1	4	22
lingvo/tasks/milan/dataset_spec.py	pip_package/setup.py	1	4	28
lingvo/core/plot.py	lingvo/core/batch_utils.py	1	30	2
lingvo/core/plot.py	lingvo/core/cluster.py	1	30	57
lingvo/core/plot.py	lingvo/core/summary_utils.py	1	30	49
lingvo/core/stream_step_test_base.py	lingvo/core/batch_major_attention.py	1	2	212
lingvo/core/stream_step_test_base.py	lingvo/core/conformer_layer.py	1	2	55
lingvo/core/stream_step_test_base.py	lingvo/core/attention_util.py	1	2	14
lingvo/core/quant_test_lib.py	lingvo/core/rnn_layers.py	1	10	72
lingvo/core/quant_test_lib.py	lingvo/core/batch_major_attention.py	1	10	212
lingvo/core/quant_test_lib.py	lingvo/core/builder_layers.py	1	10	52
lingvo/jax/tasks/lm/__init__.py	lingvo/jax/tasks/__init__.py	1	1	1
lingvo/tasks/lm/params/synthetic_packed_input.py	lingvo/core/gshard_decode.py	1	22	5
lingvo/tasks/lm/params/synthetic_packed_input.py	lingvo/core/optimizer.py	1	22	56
lingvo/tasks/lm/params/synthetic_packed_input.py	lingvo/trainer.py	1	22	248
lingvo/tasks/lm/params/synthetic_packed_input.py	lingvo/compat.py	1	22	70
lingvo/jax/tasks/lm/params/params.py	lingvo/jax/tasks/lm/params/lm_cloud.py	1	2	7
lingvo/jax/tasks/lm/params/params.py	lingvo/jax/checkpoints.py	1	2	13
lingvo/jax/tasks/lm/params/params.py	lingvo/jax/layers/__init__.py	1	2	21
lingvo/jax/tasks/lm/params/params.py	lingvo/jax/layers/transformers.py	1	2	47
lingvo/jax/tasks/lm/params/params.py	lingvo/jax/py_utils.py	1	2	12
lingvo/jax/tasks/lm/params/params.py	lingvo/jax/model.py	1	2	24
lingvo/jax/tasks/lm/params/params.py	lingvo/jax/layers/attentions.py	1	2	21
lingvo/jax/tasks/lm/params/params.py	lingvo/jax/layers/embedding_softmax.py	1	2	11
lingvo/jax/tasks/lm/params/params.py	lingvo/jax/optimizers.py	1	2	32
lingvo/jax/tasks/lm/params/params.py	lingvo/jax/eval.py	1	2	32
lingvo/jax/tasks/lm/params/params.py	lingvo/jax/trainer_lib.py	1	2	12
lingvo/jax/tasks/lm/params/params.py	lingvo/jax/model_utils.py	1	2	9
lingvo/jax/tasks/lm/params/params.py	lingvo/jax/train.py	1	2	24
lingvo/jax/tasks/lm/params/params.py	lingvo/jax/layers/convolutions.py	1	2	8
lingvo/jax/tasks/lm/params/params.py	lingvo/jax/layers/resnets.py	1	2	4
lingvo/jax/tasks/lm/params/params.py	lingvo/jax/base_layer.py	1	2	10
lingvo/jax/tasks/lm/params/params.py	lingvo/jax/layers/normalizations.py	1	2	8
lingvo/jax/tasks/lm/params/params.py	lingvo/jax/layers/recurrent.py	1	2	6
lingvo/jax/tasks/lm/params/params.py	lingvo/jax/main.py	1	2	14
lingvo/jax/tasks/lm/params/params.py	lingvo/jax/summary_utils.py	1	2	7
lingvo/jax/tasks/lm/params/params.py	lingvo/jax/layers/linears.py	1	2	7
lingvo/jax/tasks/lm/params/params.py	lingvo/jax/layers/repeats.py	1	2	5
lingvo/jax/tasks/lm/params/params.py	lingvo/jax/gshard_utils.py	1	2	4
lingvo/jax/tasks/lm/params/params.py	lingvo/jax/tasks/lm/params/bert.py	1	2	8
lingvo/jax/tasks/lm/params/params.py	lingvo/jax/base_model_params.py	1	2	4
lingvo/jax/tasks/lm/params/params.py	lingvo/jax/schedules.py	1	2	4
lingvo/jax/tasks/lm/params/params.py	lingvo/jax/layers/activations.py	1	2	4
lingvo/jax/tasks/lm/params/params.py	lingvo/jax/pytypes.py	1	2	2
lingvo/core/py_utils_flags.py	lingvo/executor.py	1	6	90
lingvo/core/py_utils_flags.py	lingvo/core/base_model.py	1	6	213
lingvo/core/py_utils_flags.py	lingvo/core/checkpointer.py	1	6	51
lingvo/core/py_utils_flags.py	lingvo/core/program.py	1	6	136
lingvo/core/py_utils_flags.py	lingvo/core/optimizer.py	1	6	56
lingvo/core/py_utils_flags.py	lingvo/core/base_layer.py	1	6	132
lingvo/core/py_utils_flags.py	lingvo/core/cluster.py	1	6	57
lingvo/jax/layers/poolings.py	lingvo/jax/base_input.py	1	3	14
lingvo/jax/layers/poolings.py	lingvo/jax/pytypes.py	1	3	2
lingvo/jax/layers/poolings.py	lingvo/jax/tasks/lm/params/params.py	1	3	2
lingvo/jax/layers/stochastics.py	lingvo/jax/base_input.py	1	3	14
lingvo/jax/layers/stochastics.py	lingvo/jax/pytypes.py	1	3	2
lingvo/jax/layers/stochastics.py	lingvo/jax/tasks/lm/params/params.py	1	3	2
lingvo/jax/layers/augmentations.py	lingvo/jax/base_input.py	1	2	14
lingvo/jax/layers/augmentations.py	lingvo/jax/pytypes.py	1	2	2
lingvo/jax/layers/augmentations.py	lingvo/jax/tasks/lm/params/params.py	1	2	2
lingvo/jax/model_imports.py	lingvo/jax/base_input.py	1	2	14
lingvo/jax/model_imports.py	lingvo/jax/pytypes.py	1	2	2
lingvo/jax/model_imports.py	lingvo/jax/tasks/lm/params/params.py	1	2	2
lingvo/jax/model_registry.py	lingvo/jax/base_input.py	1	2	14
lingvo/jax/model_registry.py	lingvo/jax/pytypes.py	1	2	2
lingvo/jax/model_registry.py	lingvo/jax/tasks/lm/params/params.py	1	2	2
lingvo/jax/train_states.py	lingvo/jax/base_input.py	1	2	14
lingvo/jax/train_states.py	lingvo/jax/pytypes.py	1	2	2
lingvo/jax/train_states.py	lingvo/jax/tasks/lm/params/params.py	1	2	2
lingvo/core/base_model_params.py	lingvo/models_test_helper.py	1	15	37
lingvo/core/recurrent.py	lingvo/core/base_layer.py	1	68	132
lingvo/core/layers_with_attention.py	lingvo/core/gshard_layers.py	1	87	52
lingvo/tasks/lm/params/wiki_bert.py	lingvo/tasks/lm/params/synthetic_packed_input.py	1	8	22
lingvo/tasks/mt/base_config.py	lingvo/tasks/mt/params/params.py	1	5	4
lingvo/core/ml_perf_log.py	lingvo/core/program.py	1	5	136
lingvo/jax/__init__.py	lingvo/jax/checkpoints.py	1	1	13
lingvo/jax/__init__.py	lingvo/jax/layers/__init__.py	1	1	21
lingvo/jax/__init__.py	lingvo/jax/layers/transformers.py	1	1	47
lingvo/jax/__init__.py	lingvo/jax/py_utils.py	1	1	12
lingvo/jax/__init__.py	lingvo/jax/model.py	1	1	24
lingvo/jax/__init__.py	lingvo/jax/layers/attentions.py	1	1	21
lingvo/jax/__init__.py	lingvo/jax/layers/embedding_softmax.py	1	1	11
lingvo/jax/__init__.py	lingvo/jax/optimizers.py	1	1	32
lingvo/jax/__init__.py	lingvo/jax/eval.py	1	1	32
lingvo/jax/__init__.py	lingvo/jax/trainer_lib.py	1	1	12
lingvo/jax/__init__.py	lingvo/jax/model_utils.py	1	1	9
lingvo/jax/__init__.py	lingvo/jax/train.py	1	1	24
lingvo/jax/__init__.py	lingvo/jax/layers/convolutions.py	1	1	8
lingvo/jax/__init__.py	lingvo/jax/layers/resnets.py	1	1	4
lingvo/jax/__init__.py	lingvo/jax/base_layer.py	1	1	10
lingvo/jax/__init__.py	lingvo/jax/layers/normalizations.py	1	1	8
lingvo/jax/__init__.py	lingvo/jax/layers/recurrent.py	1	1	6
lingvo/jax/__init__.py	lingvo/jax/main.py	1	1	14
lingvo/jax/__init__.py	lingvo/jax/summary_utils.py	1	1	7
lingvo/jax/__init__.py	lingvo/jax/layers/linears.py	1	1	7
lingvo/jax/__init__.py	lingvo/jax/layers/repeats.py	1	1	5
lingvo/jax/__init__.py	lingvo/jax/tasks/lm/input_generator.py	1	1	8
lingvo/jax/__init__.py	lingvo/jax/gshard_utils.py	1	1	4
lingvo/jax/__init__.py	lingvo/jax/tasks/lm/params/bert.py	1	1	8
lingvo/jax/__init__.py	lingvo/jax/base_model_params.py	1	1	4
lingvo/jax/__init__.py	lingvo/jax/schedules.py	1	1	4
lingvo/jax/__init__.py	lingvo/jax/layers/activations.py	1	1	4
lingvo/jax/__init__.py	lingvo/jax/pytypes.py	1	1	2
lingvo/jax/__init__.py	lingvo/jax/tasks/lm/params/params.py	1	1	2
lingvo/jax/__init__.py	lingvo/jax/layers/poolings.py	1	1	3
lingvo/jax/__init__.py	lingvo/jax/layers/stochastics.py	1	1	3
lingvo/jax/__init__.py	lingvo/jax/layers/augmentations.py	1	1	2
lingvo/jax/__init__.py	lingvo/jax/model_imports.py	1	1	2
lingvo/jax/__init__.py	lingvo/jax/model_registry.py	1	1	2
lingvo/jax/__init__.py	lingvo/jax/train_states.py	1	1	2
lingvo/jax/build-visibility.bzl	lingvo/jax/checkpoints.py	1	1	13
lingvo/jax/build-visibility.bzl	lingvo/jax/layers/__init__.py	1	1	21
lingvo/jax/build-visibility.bzl	lingvo/jax/layers/transformers.py	1	1	47
lingvo/jax/build-visibility.bzl	lingvo/jax/py_utils.py	1	1	12
lingvo/jax/build-visibility.bzl	lingvo/jax/model.py	1	1	24
lingvo/jax/build-visibility.bzl	lingvo/jax/layers/attentions.py	1	1	21
lingvo/jax/build-visibility.bzl	lingvo/jax/layers/embedding_softmax.py	1	1	11
lingvo/jax/build-visibility.bzl	lingvo/jax/optimizers.py	1	1	32
lingvo/jax/build-visibility.bzl	lingvo/jax/eval.py	1	1	32
lingvo/jax/build-visibility.bzl	lingvo/jax/trainer_lib.py	1	1	12
lingvo/jax/build-visibility.bzl	lingvo/jax/model_utils.py	1	1	9
lingvo/jax/build-visibility.bzl	lingvo/jax/train.py	1	1	24
lingvo/jax/build-visibility.bzl	lingvo/jax/layers/convolutions.py	1	1	8
lingvo/jax/build-visibility.bzl	lingvo/jax/layers/resnets.py	1	1	4
lingvo/jax/build-visibility.bzl	lingvo/jax/base_layer.py	1	1	10
lingvo/jax/build-visibility.bzl	lingvo/jax/layers/normalizations.py	1	1	8
lingvo/jax/build-visibility.bzl	lingvo/jax/layers/recurrent.py	1	1	6
lingvo/jax/build-visibility.bzl	lingvo/jax/main.py	1	1	14
lingvo/jax/build-visibility.bzl	lingvo/jax/summary_utils.py	1	1	7
lingvo/jax/build-visibility.bzl	lingvo/jax/layers/linears.py	1	1	7
lingvo/jax/build-visibility.bzl	lingvo/jax/layers/repeats.py	1	1	5
lingvo/jax/build-visibility.bzl	lingvo/jax/tasks/lm/input_generator.py	1	1	8
lingvo/jax/build-visibility.bzl	lingvo/jax/gshard_utils.py	1	1	4
lingvo/jax/build-visibility.bzl	lingvo/jax/tasks/lm/params/bert.py	1	1	8
lingvo/jax/build-visibility.bzl	lingvo/jax/base_model_params.py	1	1	4
lingvo/jax/build-visibility.bzl	lingvo/jax/schedules.py	1	1	4
lingvo/jax/build-visibility.bzl	lingvo/jax/layers/activations.py	1	1	4
lingvo/jax/build-visibility.bzl	lingvo/jax/pytypes.py	1	1	2
lingvo/jax/build-visibility.bzl	lingvo/jax/tasks/lm/params/params.py	1	1	2
lingvo/jax/build-visibility.bzl	lingvo/jax/layers/poolings.py	1	1	3
lingvo/jax/build-visibility.bzl	lingvo/jax/layers/stochastics.py	1	1	3
lingvo/jax/build-visibility.bzl	lingvo/jax/layers/augmentations.py	1	1	2
lingvo/jax/build-visibility.bzl	lingvo/jax/model_imports.py	1	1	2
lingvo/jax/build-visibility.bzl	lingvo/jax/model_registry.py	1	1	2
lingvo/jax/build-visibility.bzl	lingvo/jax/train_states.py	1	1	2
lingvo/jax/build-visibility.bzl	lingvo/jax/__init__.py	1	1	1
lingvo/jax/metric_utils.py	lingvo/jax/checkpoints.py	1	1	13
lingvo/jax/metric_utils.py	lingvo/jax/layers/__init__.py	1	1	21
lingvo/jax/metric_utils.py	lingvo/jax/layers/transformers.py	1	1	47
lingvo/jax/metric_utils.py	lingvo/jax/py_utils.py	1	1	12
lingvo/jax/metric_utils.py	lingvo/jax/model.py	1	1	24
lingvo/jax/metric_utils.py	lingvo/jax/layers/attentions.py	1	1	21
lingvo/jax/metric_utils.py	lingvo/jax/layers/embedding_softmax.py	1	1	11
lingvo/jax/metric_utils.py	lingvo/jax/optimizers.py	1	1	32
lingvo/jax/metric_utils.py	lingvo/jax/eval.py	1	1	32
lingvo/jax/metric_utils.py	lingvo/jax/trainer_lib.py	1	1	12
lingvo/jax/metric_utils.py	lingvo/jax/model_utils.py	1	1	9
lingvo/jax/metric_utils.py	lingvo/jax/train.py	1	1	24
lingvo/jax/metric_utils.py	lingvo/jax/layers/convolutions.py	1	1	8
lingvo/jax/metric_utils.py	lingvo/jax/layers/resnets.py	1	1	4
lingvo/jax/metric_utils.py	lingvo/jax/base_layer.py	1	1	10
lingvo/jax/metric_utils.py	lingvo/jax/layers/normalizations.py	1	1	8
lingvo/jax/metric_utils.py	lingvo/jax/layers/recurrent.py	1	1	6
lingvo/jax/metric_utils.py	lingvo/jax/main.py	1	1	14
lingvo/jax/metric_utils.py	lingvo/jax/summary_utils.py	1	1	7
lingvo/jax/metric_utils.py	lingvo/jax/layers/linears.py	1	1	7
lingvo/jax/metric_utils.py	lingvo/jax/layers/repeats.py	1	1	5
lingvo/jax/metric_utils.py	lingvo/jax/tasks/lm/input_generator.py	1	1	8
lingvo/jax/metric_utils.py	lingvo/jax/gshard_utils.py	1	1	4
lingvo/jax/metric_utils.py	lingvo/jax/tasks/lm/params/bert.py	1	1	8
lingvo/jax/metric_utils.py	lingvo/jax/base_model_params.py	1	1	4
lingvo/jax/metric_utils.py	lingvo/jax/schedules.py	1	1	4
lingvo/jax/metric_utils.py	lingvo/jax/layers/activations.py	1	1	4
lingvo/jax/metric_utils.py	lingvo/jax/pytypes.py	1	1	2
lingvo/jax/metric_utils.py	lingvo/jax/tasks/lm/params/params.py	1	1	2
lingvo/jax/metric_utils.py	lingvo/jax/layers/poolings.py	1	1	3
lingvo/jax/metric_utils.py	lingvo/jax/layers/stochastics.py	1	1	3
lingvo/jax/metric_utils.py	lingvo/jax/layers/augmentations.py	1	1	2
lingvo/jax/metric_utils.py	lingvo/jax/model_imports.py	1	1	2
lingvo/jax/metric_utils.py	lingvo/jax/model_registry.py	1	1	2
lingvo/jax/metric_utils.py	lingvo/jax/train_states.py	1	1	2
lingvo/jax/metric_utils.py	lingvo/jax/__init__.py	1	1	1
lingvo/jax/metric_utils.py	lingvo/jax/build-visibility.bzl	1	1	1
lingvo/tasks/lm/tools/gshard_lm_decode.py	lingvo/tasks/lm/params/synthetic_packed_input.py	1	2	22
lingvo/core/gshard_utils.py	lingvo/core/gshard_decode.py	1	12	5
lingvo/core/gshard_utils.py	lingvo/core/batch_major_attention.py	1	12	212
lingvo/tasks/asr/model.py	lingvo/core/py_utils.py	1	50	451
lingvo/tasks/asr/model.py	lingvo/tasks/lm/model.py	1	50	36
lingvo/tasks/asr/model.py	lingvo/core/layers_with_attention.py	1	50	87
lingvo/tasks/asr/model.py	lingvo/core/conformer_layer.py	1	50	55
lingvo/tasks/asr/model.py	lingvo/core/program.py	1	50	136
lingvo/core/conv_layers_with_time_padding.py	lingvo/core/batch_major_attention.py	1	46	212
lingvo/core/conv_layers_with_time_padding.py	lingvo/core/layers.py	1	46	286
lingvo/core/conv_layers_with_time_padding.py	lingvo/core/rnn_cell.py	1	46	118
lingvo/core/conv_layers_with_time_padding.py	lingvo/core/builder_layers.py	1	46	52
lingvo/core/conv_layers_with_time_padding.py	lingvo/core/attention.py	1	46	107
lingvo/core/conv_layers_with_time_padding.py	lingvo/core/quant_test_lib.py	1	46	10
lingvo/core/conv_layers_with_time_padding.py	lingvo/core/stream_step_test_base.py	1	46	2
lingvo/core/conv_layers_with_time_padding.py	lingvo/core/conformer_layer.py	1	46	55
lingvo/core/conv_layers_with_time_padding.py	lingvo/core/py_utils.py	1	46	451
lingvo/datasets.py	lingvo/models_test_helper.py	1	11	37
lingvo/tasks/lm/params/params.py	lingvo/tasks/lm/params/wiki_bert.py	1	4	8
lingvo/tasks/car/input_preprocessors.py	lingvo/tasks/car/detection_3d_lib.py	1	51	16
lingvo/pdb_wrapper.py	lingvo/executor.py	1	2	90
lingvo/pdb_wrapper.py	lingvo/base_runner.py	1	2	84
lingvo/pdb_wrapper.py	lingvo/trainer.py	1	2	248
lingvo/core/early_stop.py	lingvo/executor.py	1	14	90
lingvo/core/early_stop.py	lingvo/core/datasource.py	1	14	55
lingvo/core/early_stop.py	lingvo/core/base_input_generator.py	1	14	176
lingvo/core/early_stop.py	lingvo/trainer.py	1	14	248
lingvo/tasks/asr/eos_normalization.py	lingvo/core/batch_major_attention.py	1	3	212
lingvo/tasks/asr/eos_normalization.py	lingvo/core/py_utils.py	1	3	451
lingvo/tasks/asr/eos_normalization.py	lingvo/core/base_layer.py	1	3	132
lingvo/tasks/asr/eos_normalization.py	lingvo/lingvo.bzl	1	3	22
lingvo/tasks/asr/eos_normalization.py	lingvo/core/nested_map.py	1	3	8
lingvo/tasks/asr/eos_normalization.py	lingvo/tasks/asr/decoder_metrics.py	1	3	14
lingvo/core/hyperparams.py	lingvo/lingvo.bzl	1	64	22
lingvo/core/hyperparams.py	lingvo/core/nested_map.py	1	64	8
lingvo/core/hyperparams.py	lingvo/tasks/asr/eos_normalization.py	1	64	3
lingvo/core/hyperparams.py	lingvo/core/tokenizers.py	1	64	34
lingvo/core/ops/hyps.proto	lingvo/core/ops/x_ops.cc	1	2	55
lingvo/core/ops/hyps.proto	lingvo/core/ops/beam_search_step_op_kernels.cc	1	2	46
lingvo/core/ops/hyps.proto	lingvo/core/beam_search_helper.py	1	2	71
lingvo/core/distillation_task.py	lingvo/core/tpu_embedding_layers.py	1	4	42
lingvo/core/var_tmp_wrappers.py	lingvo/core/batch_major_attention.py	1	2	212
lingvo/core/var_tmp_wrappers.py	lingvo/core/gshard_utils.py	1	2	12
lingvo/core/var_tmp_wrappers.py	lingvo/core/py_utils.py	1	2	451
lingvo/core/var_tmp_wrappers.py	lingvo/core/base_layer.py	1	2	132
lingvo/core/var_tmp_wrappers.py	lingvo/core/recurrent.py	1	2	68
lingvo/tasks/car/waymo/tools/generate_waymo_tf.py	lingvo/core/py_utils.py	1	4	451
lingvo/tasks/car/waymo/tools/generate_waymo_tf.py	lingvo/core/predictor.py	1	4	39
lingvo/tasks/car/waymo/tools/generate_waymo_tf.py	lingvo/trainer.py	1	4	248
lingvo/tools/create_asr_features.py	lingvo/core/py_utils.py	1	16	451
lingvo/tools/create_asr_features.py	lingvo/core/predictor.py	1	16	39
lingvo/tools/create_asr_features.py	lingvo/trainer.py	1	16	248
lingvo/tools/create_asr_features.py	lingvo/tasks/car/waymo/tools/generate_waymo_tf.py	1	16	4
lingvo/core/cluster_factory.py	lingvo/core/program.py	1	22	136
lingvo/core/cluster_factory.py	lingvo/core/base_layer.py	1	22	132
lingvo/core/cluster_factory.py	lingvo/trainer.py	1	22	248
lingvo/core/cluster_factory.py	lingvo/core/datasource.py	1	22	55
lingvo/core/cluster_factory.py	lingvo/models_test_helper.py	1	22	37
lingvo/core/cluster_factory.py	lingvo/core/base_input_generator.py	1	22	176
lingvo/core/lstm_frnn_layer.py	lingvo/core/py_utils.py	1	6	451
lingvo/core/lstm_frnn_layer.py	lingvo/core/layers.py	1	6	286
lingvo/core/lstm_frnn_layer.py	lingvo/core/base_layer.py	1	6	132
lingvo/core/lstm_frnn_layer.py	lingvo/core/rnn_cell.py	1	6	118
lingvo/core/lstm_frnn_layer.py	lingvo/core/rnn_layers.py	1	6	72
lingvo/core/lstm_frnn_layer.py	lingvo/core/attention.py	1	6	107
lingvo/tasks/image/classifier.py	lingvo/core/batch_major_attention.py	1	34	212
lingvo/tasks/image/classifier.py	lingvo/core/layers.py	1	34	286
lingvo/tasks/image/input_generator.py	lingvo/tasks/image/classifier.py	1	6	34
lingvo/tasks/car/pillars.py	lingvo/tasks/car/builder_lib.py	1	19	16
lingvo/core/inference_graph.proto	lingvo/core/inference_graph_exporter.py	1	6	55
lingvo/tasks/asr/tools/simple_wer_v2.py	lingvo/tasks/asr/tools/simple_wer.py	1	4	12
lingvo/core/conv_layers_builder.py	lingvo/core/batch_major_attention.py	1	10	212
lingvo/tasks/asr/levenshtein_distance.py	lingvo/tasks/asr/decoder_utils.py	1	1	14
lingvo/core/ops/x_ops_helper.h	lingvo/core/datasource.py	1	14	55
lingvo/core/ops/x_ops_helper.h	lingvo/core/base_input_generator.py	1	14	176
lingvo/core/ops/x_ops_helper.h	lingvo/core/ops/input_common.cc	1	14	12
lingvo/core/ops/x_ops_helper.h	lingvo/core/ops/input_common.h	1	14	16
lingvo/tasks/lm/layers.py	lingvo/core/rnn_cell.py	1	43	118
lingvo/tasks/lm/layers.py	lingvo/core/batch_major_attention.py	1	43	212
lingvo/core/favor_attention.py	lingvo/core/batch_major_attention.py	1	7	212
lingvo/tasks/car/base_extractor.py	lingvo/core/tokenizers.py	1	25	34
lingvo/tasks/car/base_extractor.py	lingvo/core/hyperparams.py	1	25	64
lingvo/tasks/milan/utils.py	lingvo/tasks/milan/tf_hub_layers.py	1	2	2
lingvo/tasks/milan/utils.py	lingvo/tasks/milan/dual_encoder.py	1	2	2
lingvo/tasks/milan/utils.py	pip_package/build.Dockerfile	1	2	14
lingvo/tasks/milan/utils.py	lingvo/lingvo.bzl	1	2	22
lingvo/tasks/milan/utils.py	pip_package/setup.py	1	2	28
lingvo/core/ops/__init__.py	lingvo/core/ops/beam_search_step_op_kernels.h	1	15	16
lingvo/core/ops/__init__.py	lingvo/core/beam_search_helper.py	1	15	71
lingvo/core/ops/__init__.py	lingvo/core/ops/pack_ops.cc	1	15	9
lingvo/core/hyperparams.proto	lingvo/core/hyperparams.py	1	6	64
lingvo/tasks/milan/params/dual_encoder_recipe.py	lingvo/tasks/milan/tf_hub_layers.py	1	2	2
lingvo/tasks/milan/params/dual_encoder_recipe.py	lingvo/tasks/milan/dual_encoder.py	1	2	2
lingvo/tasks/milan/params/dual_encoder_recipe.py	pip_package/build.Dockerfile	1	2	14
lingvo/tasks/milan/params/dual_encoder_recipe.py	lingvo/lingvo.bzl	1	2	22
lingvo/tasks/milan/params/dual_encoder_recipe.py	pip_package/setup.py	1	2	28
lingvo/tasks/milan/params/dual_encoder_recipe.py	lingvo/tasks/milan/dataset_spec.py	1	2	4
lingvo/tasks/milan/params/dual_encoder_recipe.py	lingvo/tasks/milan/utils.py	1	2	2
lingvo/model_imports.py	lingvo/tasks/milan/tf_hub_layers.py	1	22	2
lingvo/model_imports.py	lingvo/tasks/milan/dual_encoder.py	1	22	2
lingvo/model_imports.py	lingvo/lingvo.bzl	1	22	22
lingvo/model_imports.py	lingvo/tasks/milan/dataset_spec.py	1	22	4
lingvo/model_imports.py	lingvo/tasks/milan/utils.py	1	22	2
lingvo/model_imports.py	lingvo/tasks/milan/params/dual_encoder_recipe.py	1	22	2
lingvo/model_imports.py	lingvo/core/datasource.py	1	22	55
lingvo/tasks/milan/labels.py	lingvo/tasks/milan/tf_hub_layers.py	1	2	2
lingvo/tasks/milan/labels.py	lingvo/tasks/milan/dual_encoder.py	1	2	2
lingvo/tasks/milan/labels.py	pip_package/build.Dockerfile	1	2	14
lingvo/tasks/milan/labels.py	lingvo/lingvo.bzl	1	2	22
lingvo/tasks/milan/labels.py	pip_package/setup.py	1	2	28
lingvo/tasks/milan/labels.py	lingvo/tasks/milan/dataset_spec.py	1	2	4
lingvo/tasks/milan/labels.py	lingvo/tasks/milan/utils.py	1	2	2
lingvo/tasks/milan/labels.py	lingvo/tasks/milan/params/dual_encoder_recipe.py	1	2	2
lingvo/tasks/milan/labels.py	lingvo/model_imports.py	1	2	22
lingvo/tasks/milan/image_preprocessor.py	lingvo/tasks/milan/tf_hub_layers.py	1	2	2
lingvo/tasks/milan/image_preprocessor.py	lingvo/tasks/milan/dual_encoder.py	1	2	2
lingvo/tasks/milan/image_preprocessor.py	pip_package/build.Dockerfile	1	2	14
lingvo/tasks/milan/image_preprocessor.py	lingvo/lingvo.bzl	1	2	22
lingvo/tasks/milan/image_preprocessor.py	pip_package/setup.py	1	2	28
lingvo/tasks/milan/image_preprocessor.py	lingvo/tasks/milan/dataset_spec.py	1	2	4
lingvo/tasks/milan/image_preprocessor.py	lingvo/tasks/milan/utils.py	1	2	2
lingvo/tasks/milan/image_preprocessor.py	lingvo/tasks/milan/params/dual_encoder_recipe.py	1	2	2
lingvo/tasks/milan/image_preprocessor.py	lingvo/model_imports.py	1	2	22
lingvo/tasks/milan/image_preprocessor.py	lingvo/tasks/milan/labels.py	1	2	2
lingvo/core/steps/embedding_steps.py	lingvo/core/rnn_cell.py	1	8	118
lingvo/core/steps/embedding_steps.py	lingvo/core/rnn_layers.py	1	8	72
lingvo/core/steps/embedding_steps.py	lingvo/tasks/lm/layers.py	1	8	43
lingvo/tasks/milan/common_schema.py	lingvo/tasks/milan/tf_hub_layers.py	1	2	2
lingvo/tasks/milan/common_schema.py	lingvo/tasks/milan/dual_encoder.py	1	2	2
lingvo/tasks/milan/common_schema.py	pip_package/build.Dockerfile	1	2	14
lingvo/tasks/milan/common_schema.py	lingvo/lingvo.bzl	1	2	22
lingvo/tasks/milan/common_schema.py	pip_package/setup.py	1	2	28
lingvo/tasks/milan/common_schema.py	lingvo/tasks/milan/dataset_spec.py	1	2	4
lingvo/tasks/milan/common_schema.py	lingvo/tasks/milan/utils.py	1	2	2
lingvo/tasks/milan/common_schema.py	lingvo/tasks/milan/params/dual_encoder_recipe.py	1	2	2
lingvo/tasks/milan/common_schema.py	lingvo/model_imports.py	1	2	22
lingvo/tasks/milan/common_schema.py	lingvo/tasks/milan/labels.py	1	2	2
lingvo/tasks/milan/common_schema.py	lingvo/tasks/milan/image_preprocessor.py	1	2	2
lingvo/core/pruning_utils.py	lingvo/core/base_model.py	1	5	213
lingvo/tasks/milan/constants.py	lingvo/tasks/milan/tf_hub_layers.py	1	1	2
lingvo/tasks/milan/constants.py	lingvo/tasks/milan/dual_encoder.py	1	1	2
lingvo/tasks/milan/constants.py	pip_package/build.Dockerfile	1	1	14
lingvo/tasks/milan/constants.py	lingvo/lingvo.bzl	1	1	22
lingvo/tasks/milan/constants.py	pip_package/setup.py	1	1	28
lingvo/tasks/milan/constants.py	lingvo/tasks/milan/dataset_spec.py	1	1	4
lingvo/tasks/milan/constants.py	lingvo/tasks/milan/utils.py	1	1	2
lingvo/tasks/milan/constants.py	lingvo/tasks/milan/params/dual_encoder_recipe.py	1	1	2
lingvo/tasks/milan/constants.py	lingvo/model_imports.py	1	1	22
lingvo/tasks/milan/constants.py	lingvo/tasks/milan/labels.py	1	1	2
lingvo/tasks/milan/constants.py	lingvo/tasks/milan/image_preprocessor.py	1	1	2
lingvo/tasks/milan/constants.py	lingvo/tasks/milan/common_schema.py	1	1	2
lingvo/tasks/milan/input_generator.py	lingvo/tasks/milan/tf_hub_layers.py	1	1	2
lingvo/tasks/milan/input_generator.py	lingvo/tasks/milan/dual_encoder.py	1	1	2
lingvo/tasks/milan/input_generator.py	pip_package/build.Dockerfile	1	1	14
lingvo/tasks/milan/input_generator.py	lingvo/lingvo.bzl	1	1	22
lingvo/tasks/milan/input_generator.py	pip_package/setup.py	1	1	28
lingvo/tasks/milan/input_generator.py	lingvo/tasks/milan/dataset_spec.py	1	1	4
lingvo/tasks/milan/input_generator.py	lingvo/tasks/milan/utils.py	1	1	2
lingvo/tasks/milan/input_generator.py	lingvo/tasks/milan/params/dual_encoder_recipe.py	1	1	2
lingvo/tasks/milan/input_generator.py	lingvo/model_imports.py	1	1	22
lingvo/tasks/milan/input_generator.py	lingvo/tasks/milan/labels.py	1	1	2
lingvo/tasks/milan/input_generator.py	lingvo/tasks/milan/image_preprocessor.py	1	1	2
lingvo/tasks/milan/input_generator.py	lingvo/tasks/milan/common_schema.py	1	1	2
lingvo/tasks/milan/input_generator.py	lingvo/tasks/milan/constants.py	1	1	1
lingvo/tasks/milan/params/cxc.py	lingvo/tasks/milan/tf_hub_layers.py	1	1	2
lingvo/tasks/milan/params/cxc.py	lingvo/tasks/milan/dual_encoder.py	1	1	2
lingvo/tasks/milan/params/cxc.py	pip_package/build.Dockerfile	1	1	14
lingvo/tasks/milan/params/cxc.py	lingvo/lingvo.bzl	1	1	22
lingvo/tasks/milan/params/cxc.py	pip_package/setup.py	1	1	28
lingvo/tasks/milan/params/cxc.py	lingvo/tasks/milan/dataset_spec.py	1	1	4
lingvo/tasks/milan/params/cxc.py	lingvo/tasks/milan/utils.py	1	1	2
lingvo/tasks/milan/params/cxc.py	lingvo/tasks/milan/params/dual_encoder_recipe.py	1	1	2
lingvo/tasks/milan/params/cxc.py	lingvo/model_imports.py	1	1	22
lingvo/tasks/milan/params/cxc.py	lingvo/tasks/milan/labels.py	1	1	2
lingvo/tasks/milan/params/cxc.py	lingvo/tasks/milan/image_preprocessor.py	1	1	2
lingvo/tasks/milan/params/cxc.py	lingvo/tasks/milan/common_schema.py	1	1	2
lingvo/tasks/milan/params/cxc.py	lingvo/tasks/milan/constants.py	1	1	1
lingvo/tasks/milan/params/cxc.py	lingvo/tasks/milan/input_generator.py	1	1	1
lingvo/tasks/milan/params/generic_datasets.py	lingvo/tasks/milan/tf_hub_layers.py	1	1	2
lingvo/tasks/milan/params/generic_datasets.py	lingvo/tasks/milan/dual_encoder.py	1	1	2
lingvo/tasks/milan/params/generic_datasets.py	pip_package/build.Dockerfile	1	1	14
lingvo/tasks/milan/params/generic_datasets.py	lingvo/lingvo.bzl	1	1	22
lingvo/tasks/milan/params/generic_datasets.py	pip_package/setup.py	1	1	28
lingvo/tasks/milan/params/generic_datasets.py	lingvo/tasks/milan/dataset_spec.py	1	1	4
lingvo/tasks/milan/params/generic_datasets.py	lingvo/tasks/milan/utils.py	1	1	2
lingvo/tasks/milan/params/generic_datasets.py	lingvo/tasks/milan/params/dual_encoder_recipe.py	1	1	2
lingvo/tasks/milan/params/generic_datasets.py	lingvo/model_imports.py	1	1	22
lingvo/tasks/milan/params/generic_datasets.py	lingvo/tasks/milan/labels.py	1	1	2
lingvo/tasks/milan/params/generic_datasets.py	lingvo/tasks/milan/image_preprocessor.py	1	1	2
lingvo/tasks/milan/params/generic_datasets.py	lingvo/tasks/milan/common_schema.py	1	1	2
lingvo/tasks/milan/params/generic_datasets.py	lingvo/tasks/milan/constants.py	1	1	1
lingvo/tasks/milan/params/generic_datasets.py	lingvo/tasks/milan/input_generator.py	1	1	1
lingvo/tasks/milan/params/generic_datasets.py	lingvo/tasks/milan/params/cxc.py	1	1	1
lingvo/tasks/milan/params/params.py	lingvo/tasks/milan/tf_hub_layers.py	1	1	2
lingvo/tasks/milan/params/params.py	lingvo/tasks/milan/dual_encoder.py	1	1	2
lingvo/tasks/milan/params/params.py	pip_package/build.Dockerfile	1	1	14
lingvo/tasks/milan/params/params.py	lingvo/lingvo.bzl	1	1	22
lingvo/tasks/milan/params/params.py	pip_package/setup.py	1	1	28
lingvo/tasks/milan/params/params.py	lingvo/tasks/milan/dataset_spec.py	1	1	4
lingvo/tasks/milan/params/params.py	lingvo/tasks/milan/utils.py	1	1	2
lingvo/tasks/milan/params/params.py	lingvo/tasks/milan/params/dual_encoder_recipe.py	1	1	2
lingvo/tasks/milan/params/params.py	lingvo/model_imports.py	1	1	22
lingvo/tasks/milan/params/params.py	lingvo/tasks/milan/labels.py	1	1	2
lingvo/tasks/milan/params/params.py	lingvo/tasks/milan/image_preprocessor.py	1	1	2
lingvo/tasks/milan/params/params.py	lingvo/tasks/milan/common_schema.py	1	1	2
lingvo/tasks/milan/params/params.py	lingvo/tasks/milan/constants.py	1	1	1
lingvo/tasks/milan/params/params.py	lingvo/tasks/milan/input_generator.py	1	1	1
lingvo/tasks/milan/params/params.py	lingvo/tasks/milan/params/cxc.py	1	1	1
lingvo/tasks/milan/params/params.py	lingvo/tasks/milan/params/generic_datasets.py	1	1	1
lingvo/tasks/milan/score_functions.py	lingvo/tasks/milan/tf_hub_layers.py	1	1	2
lingvo/tasks/milan/score_functions.py	lingvo/tasks/milan/dual_encoder.py	1	1	2
lingvo/tasks/milan/score_functions.py	pip_package/build.Dockerfile	1	1	14
lingvo/tasks/milan/score_functions.py	lingvo/lingvo.bzl	1	1	22
lingvo/tasks/milan/score_functions.py	pip_package/setup.py	1	1	28
lingvo/tasks/milan/score_functions.py	lingvo/tasks/milan/dataset_spec.py	1	1	4
lingvo/tasks/milan/score_functions.py	lingvo/tasks/milan/utils.py	1	1	2
lingvo/tasks/milan/score_functions.py	lingvo/tasks/milan/params/dual_encoder_recipe.py	1	1	2
lingvo/tasks/milan/score_functions.py	lingvo/model_imports.py	1	1	22
lingvo/tasks/milan/score_functions.py	lingvo/tasks/milan/labels.py	1	1	2
lingvo/tasks/milan/score_functions.py	lingvo/tasks/milan/image_preprocessor.py	1	1	2
lingvo/tasks/milan/score_functions.py	lingvo/tasks/milan/common_schema.py	1	1	2
lingvo/tasks/milan/score_functions.py	lingvo/tasks/milan/constants.py	1	1	1
lingvo/tasks/milan/score_functions.py	lingvo/tasks/milan/input_generator.py	1	1	1
lingvo/tasks/milan/score_functions.py	lingvo/tasks/milan/params/cxc.py	1	1	1
lingvo/tasks/milan/score_functions.py	lingvo/tasks/milan/params/generic_datasets.py	1	1	1
lingvo/tasks/milan/score_functions.py	lingvo/tasks/milan/params/params.py	1	1	1
lingvo/tasks/milan/tools/prepare_coco.py	lingvo/tasks/milan/tf_hub_layers.py	1	1	2
lingvo/tasks/milan/tools/prepare_coco.py	lingvo/tasks/milan/dual_encoder.py	1	1	2
lingvo/tasks/milan/tools/prepare_coco.py	pip_package/build.Dockerfile	1	1	14
lingvo/tasks/milan/tools/prepare_coco.py	lingvo/lingvo.bzl	1	1	22
lingvo/tasks/milan/tools/prepare_coco.py	pip_package/setup.py	1	1	28
lingvo/tasks/milan/tools/prepare_coco.py	lingvo/tasks/milan/dataset_spec.py	1	1	4
lingvo/tasks/milan/tools/prepare_coco.py	lingvo/tasks/milan/utils.py	1	1	2
lingvo/tasks/milan/tools/prepare_coco.py	lingvo/tasks/milan/params/dual_encoder_recipe.py	1	1	2
lingvo/tasks/milan/tools/prepare_coco.py	lingvo/model_imports.py	1	1	22
lingvo/tasks/milan/tools/prepare_coco.py	lingvo/tasks/milan/labels.py	1	1	2
lingvo/tasks/milan/tools/prepare_coco.py	lingvo/tasks/milan/image_preprocessor.py	1	1	2
lingvo/tasks/milan/tools/prepare_coco.py	lingvo/tasks/milan/common_schema.py	1	1	2
lingvo/tasks/milan/tools/prepare_coco.py	lingvo/tasks/milan/constants.py	1	1	1
lingvo/tasks/milan/tools/prepare_coco.py	lingvo/tasks/milan/input_generator.py	1	1	1
lingvo/tasks/milan/tools/prepare_coco.py	lingvo/tasks/milan/params/cxc.py	1	1	1
lingvo/tasks/milan/tools/prepare_coco.py	lingvo/tasks/milan/params/generic_datasets.py	1	1	1
lingvo/tasks/milan/tools/prepare_coco.py	lingvo/tasks/milan/params/params.py	1	1	1
lingvo/tasks/milan/tools/prepare_coco.py	lingvo/tasks/milan/score_functions.py	1	1	1
lingvo/tasks/milan/tpu_utils.py	lingvo/tasks/milan/tf_hub_layers.py	1	1	2
lingvo/tasks/milan/tpu_utils.py	lingvo/tasks/milan/dual_encoder.py	1	1	2
lingvo/tasks/milan/tpu_utils.py	pip_package/build.Dockerfile	1	1	14
lingvo/tasks/milan/tpu_utils.py	lingvo/lingvo.bzl	1	1	22
lingvo/tasks/milan/tpu_utils.py	pip_package/setup.py	1	1	28
lingvo/tasks/milan/tpu_utils.py	lingvo/tasks/milan/dataset_spec.py	1	1	4
lingvo/tasks/milan/tpu_utils.py	lingvo/tasks/milan/utils.py	1	1	2
lingvo/tasks/milan/tpu_utils.py	lingvo/tasks/milan/params/dual_encoder_recipe.py	1	1	2
lingvo/tasks/milan/tpu_utils.py	lingvo/model_imports.py	1	1	22
lingvo/tasks/milan/tpu_utils.py	lingvo/tasks/milan/labels.py	1	1	2
lingvo/tasks/milan/tpu_utils.py	lingvo/tasks/milan/image_preprocessor.py	1	1	2
lingvo/tasks/milan/tpu_utils.py	lingvo/tasks/milan/common_schema.py	1	1	2
lingvo/tasks/milan/tpu_utils.py	lingvo/tasks/milan/constants.py	1	1	1
lingvo/tasks/milan/tpu_utils.py	lingvo/tasks/milan/input_generator.py	1	1	1
lingvo/tasks/milan/tpu_utils.py	lingvo/tasks/milan/params/cxc.py	1	1	1
lingvo/tasks/milan/tpu_utils.py	lingvo/tasks/milan/params/generic_datasets.py	1	1	1
lingvo/tasks/milan/tpu_utils.py	lingvo/tasks/milan/params/params.py	1	1	1
lingvo/tasks/milan/tpu_utils.py	lingvo/tasks/milan/score_functions.py	1	1	1
lingvo/tasks/milan/tpu_utils.py	lingvo/tasks/milan/tools/prepare_coco.py	1	1	1
lingvo/tasks/milan/transformers.py	lingvo/tasks/milan/tf_hub_layers.py	1	1	2
lingvo/tasks/milan/transformers.py	lingvo/tasks/milan/dual_encoder.py	1	1	2
lingvo/tasks/milan/transformers.py	pip_package/build.Dockerfile	1	1	14
lingvo/tasks/milan/transformers.py	lingvo/lingvo.bzl	1	1	22
lingvo/tasks/milan/transformers.py	pip_package/setup.py	1	1	28
lingvo/tasks/milan/transformers.py	lingvo/tasks/milan/dataset_spec.py	1	1	4
lingvo/tasks/milan/transformers.py	lingvo/tasks/milan/utils.py	1	1	2
lingvo/tasks/milan/transformers.py	lingvo/tasks/milan/params/dual_encoder_recipe.py	1	1	2
lingvo/tasks/milan/transformers.py	lingvo/model_imports.py	1	1	22
lingvo/tasks/milan/transformers.py	lingvo/tasks/milan/labels.py	1	1	2
lingvo/tasks/milan/transformers.py	lingvo/tasks/milan/image_preprocessor.py	1	1	2
lingvo/tasks/milan/transformers.py	lingvo/tasks/milan/common_schema.py	1	1	2
lingvo/tasks/milan/transformers.py	lingvo/tasks/milan/constants.py	1	1	1
lingvo/tasks/milan/transformers.py	lingvo/tasks/milan/input_generator.py	1	1	1
lingvo/tasks/milan/transformers.py	lingvo/tasks/milan/params/cxc.py	1	1	1
lingvo/tasks/milan/transformers.py	lingvo/tasks/milan/params/generic_datasets.py	1	1	1
lingvo/tasks/milan/transformers.py	lingvo/tasks/milan/params/params.py	1	1	1
lingvo/tasks/milan/transformers.py	lingvo/tasks/milan/score_functions.py	1	1	1
lingvo/tasks/milan/transformers.py	lingvo/tasks/milan/tools/prepare_coco.py	1	1	1
lingvo/tasks/milan/transformers.py	lingvo/tasks/milan/tpu_utils.py	1	1	1
lingvo/core/scatter_update.py	lingvo/core/batch_major_attention.py	1	1	212
lingvo/core/entmax.py	lingvo/core/gshard_builder.py	1	2	86