Path	Lines of Code
floyd.yml	2
setup.py	93
tensor2tensor/__init__.py	1
tensor2tensor/data_generators/__init__.py	1
tensor2tensor/data_generators/algorithmic.py	278
tensor2tensor/data_generators/algorithmic_math.py	252
tensor2tensor/data_generators/algorithmic_math_deepmind.py	54
tensor2tensor/data_generators/algorithmic_math_two_variables.py	55
tensor2tensor/data_generators/all_problems.py	121
tensor2tensor/data_generators/allen_brain.py	247
tensor2tensor/data_generators/audio.py	94
tensor2tensor/data_generators/audio_encoder.py	42
tensor2tensor/data_generators/babi_qa.py	291
tensor2tensor/data_generators/bair_robot_pushing.py	129
tensor2tensor/data_generators/celeba.py	175
tensor2tensor/data_generators/celebahq.py	69
tensor2tensor/data_generators/cifar.py	442
tensor2tensor/data_generators/cipher.py	120
tensor2tensor/data_generators/cleaner_en_xx.py	121
tensor2tensor/data_generators/cnn_dailymail.py	249
tensor2tensor/data_generators/cola.py	73
tensor2tensor/data_generators/common_voice.py	149
tensor2tensor/data_generators/conll_ner.py	56
tensor2tensor/data_generators/desc2code.py	173
tensor2tensor/data_generators/dialog_abstract.py	247
tensor2tensor/data_generators/dialog_cornell.py	99
tensor2tensor/data_generators/dialog_dailydialog.py	73
tensor2tensor/data_generators/dialog_opensubtitles.py	140
tensor2tensor/data_generators/dialog_personachat.py	122
tensor2tensor/data_generators/dna_encoder.py	79
tensor2tensor/data_generators/enwik8.py	114
tensor2tensor/data_generators/fsns.py	54
tensor2tensor/data_generators/function_docstring.py	57
tensor2tensor/data_generators/gene_expression.py	188
tensor2tensor/data_generators/generator_utils.py	665
tensor2tensor/data_generators/google_robot_pushing.py	86
tensor2tensor/data_generators/gym_env.py	605
tensor2tensor/data_generators/ice_parsing.py	76
tensor2tensor/data_generators/image_lsun.py	63
tensor2tensor/data_generators/image_utils.py	255
tensor2tensor/data_generators/imagenet.py	356
tensor2tensor/data_generators/imdb.py	67
tensor2tensor/data_generators/inspect_tfrecord.py	63
tensor2tensor/data_generators/lambada.py	176
tensor2tensor/data_generators/librispeech.py	208
tensor2tensor/data_generators/lm1b.py	104
tensor2tensor/data_generators/lm1b_imdb.py	17
tensor2tensor/data_generators/lm1b_mnli.py	27
tensor2tensor/data_generators/mnist.py	138
tensor2tensor/data_generators/moving_mnist.py	101
tensor2tensor/data_generators/mrpc.py	95
tensor2tensor/data_generators/mscoco.py	178
tensor2tensor/data_generators/multi_problem.py	342
tensor2tensor/data_generators/multi_problem_v2.py	169
tensor2tensor/data_generators/multinli.py	135
tensor2tensor/data_generators/ocr.py	52
tensor2tensor/data_generators/ops/pack_sequences_ops.cc	525
tensor2tensor/data_generators/ops/subword_text_encoder.cc	102
tensor2tensor/data_generators/ops/subword_text_encoder.h	21
tensor2tensor/data_generators/ops/subword_text_encoder_ops.cc	53
tensor2tensor/data_generators/paraphrase_ms_coco.py	120
tensor2tensor/data_generators/pointer_generator_word.py	102
tensor2tensor/data_generators/problem.py	593
tensor2tensor/data_generators/problem_hparams.py	143
tensor2tensor/data_generators/program_search.py	65
tensor2tensor/data_generators/ptb.py	98
tensor2tensor/data_generators/qnli.py	77
tensor2tensor/data_generators/quora_qpairs.py	82
tensor2tensor/data_generators/rte.py	77
tensor2tensor/data_generators/scitail.py	81
tensor2tensor/data_generators/seq2edits.py	164
tensor2tensor/data_generators/snli.py	103
tensor2tensor/data_generators/speech_recognition.py	97
tensor2tensor/data_generators/squad.py	159
tensor2tensor/data_generators/sst_binary.py	74
tensor2tensor/data_generators/stanford_nli.py	94
tensor2tensor/data_generators/style_transfer.py	98
tensor2tensor/data_generators/subject_verb_agreement.py	152
tensor2tensor/data_generators/text_encoder.py	525
tensor2tensor/data_generators/text_encoder_build_subword.py	40
tensor2tensor/data_generators/text_problems.py	752
tensor2tensor/data_generators/timeseries.py	261
tensor2tensor/data_generators/timeseries_data_generator.py	14
tensor2tensor/data_generators/tokenizer.py	82
tensor2tensor/data_generators/transduction_problems.py	141
tensor2tensor/data_generators/translate.py	261
tensor2tensor/data_generators/translate_encs.py	67
tensor2tensor/data_generators/translate_encs_cubbitt.py	50
tensor2tensor/data_generators/translate_ende.py	152
tensor2tensor/data_generators/translate_enes.py	79
tensor2tensor/data_generators/translate_enet.py	44
tensor2tensor/data_generators/translate_enfr.py	185
tensor2tensor/data_generators/translate_enid.py	46
tensor2tensor/data_generators/translate_enmk.py	33
tensor2tensor/data_generators/translate_enro.py	110
tensor2tensor/data_generators/translate_entn.py	25
tensor2tensor/data_generators/translate_envi.py	24
tensor2tensor/data_generators/translate_enzh.py	181
tensor2tensor/data_generators/video_generated.py	132
tensor2tensor/data_generators/video_utils.py	487
tensor2tensor/data_generators/vqa.py	359
tensor2tensor/data_generators/vqa_utils.py	106
tensor2tensor/data_generators/wiki.py	219
tensor2tensor/data_generators/wiki_lm.py	178
tensor2tensor/data_generators/wiki_multi_problems.py	252
tensor2tensor/data_generators/wiki_revision.py	362
tensor2tensor/data_generators/wiki_revision_utils.py	335
tensor2tensor/data_generators/wikisum/__init__.py	1
tensor2tensor/data_generators/wikisum/generate_vocab.py	24
tensor2tensor/data_generators/wikisum/get_references_commoncrawl.py	41
tensor2tensor/data_generators/wikisum/get_references_web.py	38
tensor2tensor/data_generators/wikisum/get_references_web_single_group.py	210
tensor2tensor/data_generators/wikisum/html.py	40
tensor2tensor/data_generators/wikisum/parallel_launch.py	199
tensor2tensor/data_generators/wikisum/produce_examples.py	44
tensor2tensor/data_generators/wikisum/utils.py	161
tensor2tensor/data_generators/wikisum/validate_data.py	125
tensor2tensor/data_generators/wikisum/wikisum.py	397
tensor2tensor/data_generators/wikitext103.py	123
tensor2tensor/data_generators/wnli.py	78
tensor2tensor/data_generators/wsj_parsing.py	73
tensor2tensor/data_generators/yelp_full.py	67
tensor2tensor/data_generators/yelp_polarity.py	67
tensor2tensor/envs/__init__.py	6
tensor2tensor/envs/env_problem.py	289
tensor2tensor/envs/env_problem_utils.py	147
tensor2tensor/envs/gym_env_problem.py	140
tensor2tensor/envs/gym_spaces_utils.py	41
tensor2tensor/envs/mujoco_problems.py	42
tensor2tensor/envs/rendered_env_problem.py	74
tensor2tensor/envs/tic_tac_toe_env.py	135
tensor2tensor/envs/tic_tac_toe_env_problem.py	27
tensor2tensor/envs/time_step.py	21
tensor2tensor/envs/trajectory.py	300
tensor2tensor/insights/__init__.py	1
tensor2tensor/insights/graph.py	52
tensor2tensor/insights/insight_configuration.proto	47
tensor2tensor/insights/polymer/attention_visualization/attention-visualization.html	94
tensor2tensor/insights/polymer/attention_visualization/attention-visualization.js	204
tensor2tensor/insights/polymer/common-types.js	14
tensor2tensor/insights/polymer/explore_view/explore-view.html	116
tensor2tensor/insights/polymer/explore_view/explore-view.js	97
tensor2tensor/insights/polymer/graph_visualization/graph-visualization.html	146
tensor2tensor/insights/polymer/graph_visualization/graph-visualization.js	536
tensor2tensor/insights/polymer/index.html	49
tensor2tensor/insights/polymer/insights_app/insights-app.html	92
tensor2tensor/insights/polymer/insights_app/insights-app.js	30
tensor2tensor/insights/polymer/language_selector/language-selector-content.html	40
tensor2tensor/insights/polymer/language_selector/language-selector-content.js	125
tensor2tensor/insights/polymer/language_selector/language-selector.html	22
tensor2tensor/insights/polymer/language_selector/language-selector.js	27
tensor2tensor/insights/polymer/processing_visualization/processing-visualization.html	56
tensor2tensor/insights/polymer/processing_visualization/processing-visualization.js	13
tensor2tensor/insights/polymer/query_card/query-card.html	60
tensor2tensor/insights/polymer/query_card/query-card.js	169
tensor2tensor/insights/polymer/tensor2tensor.html	49
tensor2tensor/insights/polymer/translation_result/translation-result.html	60
tensor2tensor/insights/polymer/translation_result/translation-result.js	41
tensor2tensor/insights/query_processor.py	4
tensor2tensor/insights/server.py	108
tensor2tensor/insights/transformer_model.py	163
tensor2tensor/layers/__init__.py	1
tensor2tensor/layers/area_attention.py	285
tensor2tensor/layers/common_attention.py	3314
tensor2tensor/layers/common_audio.py	65
tensor2tensor/layers/common_hparams.py	256
tensor2tensor/layers/common_image_attention.py	529
tensor2tensor/layers/common_layers.py	2320
tensor2tensor/layers/common_video.py	491
tensor2tensor/layers/discretization.py	951
tensor2tensor/layers/latent_layers.py	416
tensor2tensor/layers/message_passing_attention.py	386
tensor2tensor/layers/modalities.py	1088
tensor2tensor/layers/ngram.py	40
tensor2tensor/layers/transformer_glow_layers.py	293
tensor2tensor/layers/transformer_glow_layers_ops.py	181
tensor2tensor/layers/transformer_layers.py	282
tensor2tensor/layers/transformer_memory.py	206
tensor2tensor/layers/vq_discrete.py	185
tensor2tensor/layers/vqa_layers.py	219
tensor2tensor/metrics/__init__.py	1
tensor2tensor/metrics/video_conditional_fvd.py	16
tensor2tensor/models/__init__.py	69
tensor2tensor/models/basic.py	32
tensor2tensor/models/bytenet.py	74
tensor2tensor/models/distillation.py	117
tensor2tensor/models/evolved_transformer.py	586
tensor2tensor/models/image_transformer.py	835
tensor2tensor/models/image_transformer_2d.py	692
tensor2tensor/models/lstm.py	340
tensor2tensor/models/mtf_image_transformer.py	479
tensor2tensor/models/mtf_resnet.py	284
tensor2tensor/models/mtf_transformer.py	903
tensor2tensor/models/mtf_transformer2.py	607
tensor2tensor/models/neural_architecture_search/__init__.py	1
tensor2tensor/models/neural_architecture_search/nas_layers.py	447
tensor2tensor/models/neural_architecture_search/nas_model.py	644
tensor2tensor/models/neural_assistant.py	333
tensor2tensor/models/neural_gpu.py	80
tensor2tensor/models/research/__init__.py	1
tensor2tensor/models/research/adafactor_experiments.py	148
tensor2tensor/models/research/aligned.py	317
tensor2tensor/models/research/attention_lm.py	125
tensor2tensor/models/research/attention_lm_moe.py	518
tensor2tensor/models/research/autoencoders.py	1073
tensor2tensor/models/research/cycle_gan.py	97
tensor2tensor/models/research/gene_expression.py	92
tensor2tensor/models/research/glow.py	115
tensor2tensor/models/research/glow_init_hook.py	12
tensor2tensor/models/research/glow_ops.py	713
tensor2tensor/models/research/lm_experiments.py	83
tensor2tensor/models/research/moe.py	259
tensor2tensor/models/research/moe_experiments.py	314
tensor2tensor/models/research/multiquery_paper.py	152
tensor2tensor/models/research/neural_stack.py	321
tensor2tensor/models/research/residual_shuffle_exchange.py	124
tensor2tensor/models/research/rl.py	625
tensor2tensor/models/research/shuffle_network.py	216
tensor2tensor/models/research/similarity_transformer.py	60
tensor2tensor/models/research/super_lm.py	272
tensor2tensor/models/research/transformer_aux.py	70
tensor2tensor/models/research/transformer_moe.py	229
tensor2tensor/models/research/transformer_nat.py	328
tensor2tensor/models/research/transformer_parallel.py	231
tensor2tensor/models/research/transformer_revnet.py	144
tensor2tensor/models/research/transformer_seq2edits.py	357
tensor2tensor/models/research/transformer_sketch.py	32
tensor2tensor/models/research/transformer_symshard.py	307
tensor2tensor/models/research/transformer_vae.py	826
tensor2tensor/models/research/transformer_vae_flow_prior.py	826
tensor2tensor/models/research/transformer_vae_flow_prior_ops.py	278
tensor2tensor/models/research/universal_transformer.py	456
tensor2tensor/models/research/universal_transformer_util.py	745
tensor2tensor/models/research/vqa_attention.py	400
tensor2tensor/models/research/vqa_recurrent_self_attention.py	191
tensor2tensor/models/research/vqa_self_attention.py	603
tensor2tensor/models/resnet.py	542
tensor2tensor/models/revnet.py	215
tensor2tensor/models/shake_shake.py	160
tensor2tensor/models/slicenet.py	286
tensor2tensor/models/text_cnn.py	66
tensor2tensor/models/transformer.py	1978
tensor2tensor/models/vanilla_gan.py	135
tensor2tensor/models/video/__init__.py	1
tensor2tensor/models/video/base.py	394
tensor2tensor/models/video/base_vae.py	66
tensor2tensor/models/video/basic_deterministic.py	161
tensor2tensor/models/video/basic_deterministic_params.py	136
tensor2tensor/models/video/basic_recurrent.py	33
tensor2tensor/models/video/basic_stochastic.py	240
tensor2tensor/models/video/emily.py	337
tensor2tensor/models/video/epva.py	493
tensor2tensor/models/video/epva_params.py	41
tensor2tensor/models/video/next_frame_glow.py	383
tensor2tensor/models/video/nfg_interpolate.py	159
tensor2tensor/models/video/nfg_test_utils.py	130
tensor2tensor/models/video/savp.py	318
tensor2tensor/models/video/savp_params.py	62
tensor2tensor/models/video/sv2p.py	575
tensor2tensor/models/video/sv2p_params.py	107
tensor2tensor/models/xception.py	135
tensor2tensor/problems.py	10
tensor2tensor/problems_colab.py	11
tensor2tensor/rl/__init__.py	1
tensor2tensor/rl/datagen_with_agent.py	34
tensor2tensor/rl/dopamine_connector.py	519
tensor2tensor/rl/envs/__init__.py	1
tensor2tensor/rl/envs/in_graph_batch_env.py	46
tensor2tensor/rl/envs/py_func_batch_env.py	57
tensor2tensor/rl/envs/simulated_batch_env.py	208
tensor2tensor/rl/envs/simulated_batch_gym_env.py	58
tensor2tensor/rl/envs/tf_atari_wrappers.py	75
tensor2tensor/rl/evaluator.py	433
tensor2tensor/rl/gym_utils.py	194
tensor2tensor/rl/player.py	243
tensor2tensor/rl/player_utils.py	185
tensor2tensor/rl/policy_learner.py	27
tensor2tensor/rl/ppo.py	172
tensor2tensor/rl/ppo_learner.py	410
tensor2tensor/rl/restarter.py	63
tensor2tensor/rl/rl_utils.py	609
tensor2tensor/rl/trainer_model_based.py	260
tensor2tensor/rl/trainer_model_based_agent_only.py	38
tensor2tensor/rl/trainer_model_based_params.py	655
tensor2tensor/rl/trainer_model_free.py	96
tensor2tensor/serving/__init__.py	1
tensor2tensor/serving/export.py	138
tensor2tensor/serving/query.py	72
tensor2tensor/serving/serving_utils.py	109
tensor2tensor/utils/__init__.py	1
tensor2tensor/utils/adafactor.py	165
tensor2tensor/utils/adv_attack_utils.py	135
tensor2tensor/utils/avg_checkpoints.py	82
tensor2tensor/utils/beam_search.py	320
tensor2tensor/utils/bleu_hook.py	156
tensor2tensor/utils/cloud_mlengine.py	258
tensor2tensor/utils/compute_video_metrics.py	27
tensor2tensor/utils/contrib.py	140
tensor2tensor/utils/data_reader.py	360
tensor2tensor/utils/decoding.py	748
tensor2tensor/utils/devices.py	115
tensor2tensor/utils/diet.py	203
tensor2tensor/utils/expert_utils.py	655
tensor2tensor/utils/flags.py	97
tensor2tensor/utils/get_rouge.py	52
tensor2tensor/utils/hparam.py	211
tensor2tensor/utils/hparams_lib.py	65
tensor2tensor/utils/learning_rate.py	144
tensor2tensor/utils/metrics.py	516
tensor2tensor/utils/metrics_hook.py	160
tensor2tensor/utils/misc_utils.py	14
tensor2tensor/utils/mlperf_log.py	76
tensor2tensor/utils/mlperf_tags.py	124
tensor2tensor/utils/mtf_model.py	202
tensor2tensor/utils/multistep_optimizer.py	86
tensor2tensor/utils/multistep_with_adamoptimizer.py	176
tensor2tensor/utils/optimize.py	288
tensor2tensor/utils/partial_checkpoint_load_hook.py	17
tensor2tensor/utils/pruning_utils.py	44
tensor2tensor/utils/quantization.py	103
tensor2tensor/utils/registry.py	246
tensor2tensor/utils/restore_hook.py	27
tensor2tensor/utils/rouge.py	77
tensor2tensor/utils/sari_hook.py	114
tensor2tensor/utils/scheduled_sampling.py	121
tensor2tensor/utils/t2t_model.py	1527
tensor2tensor/utils/trainer_lib.py	681
tensor2tensor/utils/update_ops_hook.py	9
tensor2tensor/utils/usr_dir.py	21
tensor2tensor/utils/video/prediction2gif.py	134
tensor2tensor/utils/video/reward_confusion.py	59
tensor2tensor/utils/video2gif.py	48
tensor2tensor/utils/video_metrics.py	144
tensor2tensor/utils/yellowfin.py	318
tensor2tensor/visualization/__init__.py	1
tensor2tensor/visualization/attention.js	307
tensor2tensor/visualization/attention.py	80
tensor2tensor/visualization/visualization.py	88