tensor2tensor/layers/modalities.py
tensor2tensor/models/xception.py

tensor2tensor/layers/transformer_layers.py
tensor2tensor/models/evolved_transformer.py

tensor2tensor/layers/transformer_layers.py
tensor2tensor/models/transformer.py

tensor2tensor/layers/transformer_layers.py
tensor2tensor/models/research/universal_transformer_util.py

tensor2tensor/layers/modalities.py
tensor2tensor/models/research/transformer_vae_flow_prior_ops.py

tensor2tensor/layers/transformer_layers.py
tensor2tensor/models/research/vqa_self_attention.py

tensor2tensor/layers/common_image_attention.py
tensor2tensor/models/research/aligned.py

tensor2tensor/layers/discretization.py
tensor2tensor/models/research/transformer_nat.py

tensor2tensor/layers/common_image_attention.py
tensor2tensor/models/research/vqa_self_attention.py

tensor2tensor/layers/common_layers.py
tensor2tensor/models/resnet.py

tensor2tensor/layers/common_image_attention.py
tensor2tensor/models/research/universal_transformer_util.py

tensor2tensor/layers/common_image_attention.py
tensor2tensor/models/research/attention_lm_moe.py

tensor2tensor/layers/common_image_attention.py
tensor2tensor/models/evolved_transformer.py

tensor2tensor/layers/latent_layers.py
tensor2tensor/models/research/transformer_nat.py

tensor2tensor/layers/common_image_attention.py
tensor2tensor/models/transformer.py