id: 1 unit: def tpu_estimator_model_fn() file: mesh_tensorflow/transformer/utils.py start line: 0 end line: 0 size: 431 LOC McCabe index: 80 number of parameters: 26 id: 2 unit: def beam_search() file: mesh_tensorflow/beam_search.py start line: 0 end line: 0 size: 214 LOC McCabe index: 19 number of parameters: 12 id: 3 unit: def transformer_moe_layer_v1() file: mesh_tensorflow/transformer/heterogeneous_moe.py start line: 0 end line: 0 size: 204 LOC McCabe index: 31 number of parameters: 12 id: 4 unit: def run() file: mesh_tensorflow/transformer/utils.py start line: 0 end line: 0 size: 202 LOC McCabe index: 32 number of parameters: 21 id: 5 unit: def transformer_moe_layer_v1() file: mesh_tensorflow/transformer/moe.py start line: 0 end line: 0 size: 187 LOC McCabe index: 27 number of parameters: 11 id: 6 unit: def unet_with_spatial_partition() file: mesh_tensorflow/experimental/unet.py start line: 0 end line: 0 size: 185 LOC McCabe index: 18 number of parameters: 5 id: 7 unit: def model_fn_builder() file: mesh_tensorflow/bert/run_pretraining.py start line: 0 end line: 0 size: 169 LOC McCabe index: 10 number of parameters: 5 id: 8 unit: def eval_model() file: mesh_tensorflow/transformer/utils.py start line: 0 end line: 0 size: 150 LOC McCabe index: 38 number of parameters: 13 id: 9 unit: def main() file: mesh_tensorflow/bert/run_classifier.py start line: 0 end line: 0 size: 150 LOC McCabe index: 31 number of parameters: 1 id: 10 unit: def write_predictions() file: mesh_tensorflow/bert/run_squad.py start line: 0 end line: 0 size: 147 LOC McCabe index: 34 number of parameters: 9 id: 11 unit: def model_fn_builder() file: mesh_tensorflow/bert/run_classifier.py start line: 0 end line: 0 size: 143 LOC McCabe index: 14 number of parameters: 7 id: 12 unit: def synthetic_attention() file: mesh_tensorflow/transformer/attention.py start line: 0 end line: 0 size: 142 LOC McCabe index: 20 number of parameters: 15 id: 13 unit: def sample_autoregressive() file: mesh_tensorflow/transformer/transformer.py start line: 0 end line: 0 size: 136 LOC McCabe index: 18 number of parameters: 6 id: 14 unit: def _top_2_gating() file: mesh_tensorflow/transformer/moe.py start line: 0 end line: 0 size: 136 LOC McCabe index: 28 number of parameters: 11 id: 15 unit: def model_fn_builder() file: mesh_tensorflow/bert/run_squad.py start line: 0 end line: 0 size: 136 LOC McCabe index: 13 number of parameters: 6 id: 16 unit: def convert_examples_to_features() file: mesh_tensorflow/bert/run_squad.py start line: 0 end line: 0 size: 134 LOC McCabe index: 41 number of parameters: 7 id: 17 unit: def main() file: mesh_tensorflow/bert/run_squad.py start line: 0 end line: 0 size: 123 LOC McCabe index: 12 number of parameters: 1 id: 18 unit: def get_dataset_creator() file: mesh_tensorflow/experimental/unet.py start line: 0 end line: 0 size: 120 LOC McCabe index: 17 number of parameters: 1 id: 19 unit: def _ntlb_gating() file: mesh_tensorflow/transformer/moe.py start line: 0 end line: 0 size: 119 LOC McCabe index: 17 number of parameters: 11 id: 20 unit: def _top_n_gating() file: mesh_tensorflow/transformer/moe.py start line: 0 end line: 0 size: 113 LOC McCabe index: 35 number of parameters: 11 id: 21 unit: def __init__() file: mesh_tensorflow/bert/bert.py start line: 0 end line: 0 size: 112 LOC McCabe index: 19 number of parameters: 9 id: 22 unit: def transformer_moe_layer_v2() file: mesh_tensorflow/transformer/moe.py start line: 0 end line: 0 size: 107 LOC McCabe index: 23 number of parameters: 9 id: 23 unit: def _switch_gating() file: mesh_tensorflow/transformer/moe.py start line: 0 end line: 0 size: 104 LOC McCabe index: 21 number of parameters: 11 id: 24 unit: def beam_search() file: mesh_tensorflow/transformer/transformer.py start line: 0 end line: 0 size: 95 LOC McCabe index: 8 number of parameters: 4 id: 25 unit: def _switch_max_gating() file: mesh_tensorflow/transformer/moe.py start line: 0 end line: 0 size: 95 LOC McCabe index: 22 number of parameters: 11 id: 26 unit: def _get_model_fn() file: mesh_tensorflow/experimental/model_executor.py start line: 0 end line: 0 size: 93 LOC McCabe index: 19 number of parameters: 2 id: 27 unit: def __init__() file: mesh_tensorflow/transformer/attention.py start line: 0 end line: 0 size: 90 LOC McCabe index: 12 number of parameters: 19 id: 28 unit: def call() file: mesh_tensorflow/transformer/transformer_layers.py start line: 0 end line: 0 size: 89 LOC McCabe index: 10 number of parameters: 4 id: 29 unit: def main() file: mesh_tensorflow/bert/run_pretraining.py start line: 0 end line: 0 size: 88 LOC McCabe index: 19 number of parameters: 1 id: 30 unit: def __init__() file: mesh_tensorflow/transformer/vocab_embeddings.py start line: 0 end line: 0 size: 83 LOC McCabe index: 2 number of parameters: 12 id: 31 unit: def _expert_selection_gating() file: mesh_tensorflow/transformer/moe.py start line: 0 end line: 0 size: 81 LOC McCabe index: 21 number of parameters: 13 id: 32 unit: def rewrite_stack_variables() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 79 LOC McCabe index: 26 number of parameters: 4 id: 33 unit: def _eval_phase() file: mesh_tensorflow/experimental/model_executor.py start line: 0 end line: 0 size: 75 LOC McCabe index: 6 number of parameters: 3 id: 34 unit: def _pack_with_tf_ops() file: mesh_tensorflow/transformer/dataset.py start line: 0 end line: 0 size: 75 LOC McCabe index: 11 number of parameters: 3 id: 35 unit: def act_layer() file: mesh_tensorflow/transformer/universal_transformer.py start line: 0 end line: 0 size: 75 LOC McCabe index: 12 number of parameters: 4 id: 36 unit: def _dynamic_text2self() file: mesh_tensorflow/transformer/utils.py start line: 0 end line: 0 size: 74 LOC McCabe index: 6 number of parameters: 1 id: 37 unit: def decode() file: mesh_tensorflow/transformer/funnel_transformer.py start line: 0 end line: 0 size: 74 LOC McCabe index: 5 number of parameters: 3 id: 38 unit: def decode() file: mesh_tensorflow/transformer/transformer.py start line: 0 end line: 0 size: 73 LOC McCabe index: 5 number of parameters: 3 id: 39 unit: def _call_internal() file: mesh_tensorflow/transformer/transformer.py start line: 0 end line: 0 size: 72 LOC McCabe index: 18 number of parameters: 4 id: 40 unit: def compute_bias() file: mesh_tensorflow/transformer/transformer_layers.py start line: 0 end line: 0 size: 69 LOC McCabe index: 19 number of parameters: 6 id: 41 unit: def apply_grad() file: mesh_tensorflow/optimize.py start line: 0 end line: 0 size: 67 LOC McCabe index: 11 number of parameters: 3 id: 42 unit: def call_simple() file: mesh_tensorflow/transformer/transformer.py start line: 0 end line: 0 size: 67 LOC McCabe index: 10 number of parameters: 6 id: 43 unit: def read_squad_examples() file: mesh_tensorflow/bert/run_squad.py start line: 0 end line: 0 size: 67 LOC McCabe index: 19 number of parameters: 2 id: 44 unit: def get_checkpoint_iterator() file: mesh_tensorflow/transformer/utils.py start line: 0 end line: 0 size: 66 LOC McCabe index: 5 number of parameters: 5 id: 45 unit: def get_estimator() file: mesh_tensorflow/transformer/utils.py start line: 0 end line: 0 size: 63 LOC McCabe index: 18 number of parameters: 22 id: 46 unit: def __init__() file: mesh_tensorflow/simd_mesh_impl.py start line: 0 end line: 0 size: 62 LOC McCabe index: 10 number of parameters: 3 id: 47 unit: def __init__() file: mesh_tensorflow/transformer/heterogeneous_moe.py start line: 0 end line: 0 size: 62 LOC McCabe index: 1 number of parameters: 30 id: 48 unit: def local_attention_1d() file: mesh_tensorflow/transformer/attention.py start line: 0 end line: 0 size: 62 LOC McCabe index: 11 number of parameters: 13 id: 49 unit: def convert_single_example() file: mesh_tensorflow/bert/run_classifier.py start line: 0 end line: 0 size: 62 LOC McCabe index: 27 number of parameters: 5 id: 50 unit: def __init__() file: mesh_tensorflow/transformer/transformer.py start line: 0 end line: 0 size: 61 LOC McCabe index: 8 number of parameters: 22 id: 51 unit: def __init__() file: mesh_tensorflow/transformer/moe.py start line: 0 end line: 0 size: 60 LOC McCabe index: 1 number of parameters: 29 id: 52 unit: def __init__() file: mesh_tensorflow/transformer/transformer.py start line: 0 end line: 0 size: 59 LOC McCabe index: 3 number of parameters: 27 id: 53 unit: def _add_constraints() file: mesh_tensorflow/auto_mtf/layout_optimizer.py start line: 0 end line: 0 size: 58 LOC McCabe index: 21 number of parameters: 1 id: 54 unit: def create_optimizer() file: mesh_tensorflow/bert/optimization.py start line: 0 end line: 0 size: 56 LOC McCabe index: 13 number of parameters: 7 id: 55 unit: def _minimize_peak_memory_list() file: mesh_tensorflow/auto_mtf/scheduler.py start line: 0 end line: 0 size: 54 LOC McCabe index: 25 number of parameters: 1 id: 56 unit: def _train_phase() file: mesh_tensorflow/experimental/model_executor.py start line: 0 end line: 0 size: 54 LOC McCabe index: 9 number of parameters: 3 id: 57 unit: def dense() file: mesh_tensorflow/layers.py start line: 0 end line: 0 size: 53 LOC McCabe index: 13 number of parameters: 12 id: 58 unit: def _gen_infeed_queue() file: mesh_tensorflow/experimental/input_reader.py start line: 0 end line: 0 size: 53 LOC McCabe index: 14 number of parameters: 1 id: 59 unit: def call_simple() file: mesh_tensorflow/transformer/transformer.py start line: 0 end line: 0 size: 53 LOC McCabe index: 6 number of parameters: 5 id: 60 unit: def masked_local_attention_1d() file: mesh_tensorflow/layers.py start line: 0 end line: 0 size: 52 LOC McCabe index: 12 number of parameters: 11 id: 61 unit: def call_simple() file: mesh_tensorflow/transformer/funnel_transformer.py start line: 0 end line: 0 size: 52 LOC McCabe index: 7 number of parameters: 6 id: 62 unit: def __init__() file: mesh_tensorflow/bert/bert.py start line: 0 end line: 0 size: 52 LOC McCabe index: 3 number of parameters: 25 id: 63 unit: def maybe_gen_fake_data_based_on_real_data() file: mesh_tensorflow/experimental/data_aug_lib.py start line: 0 end line: 0 size: 51 LOC McCabe index: 3 number of parameters: 5 id: 64 unit: def call_simple() file: mesh_tensorflow/transformer/transformer.py start line: 0 end line: 0 size: 51 LOC McCabe index: 8 number of parameters: 6 id: 65 unit: def __init__() file: mesh_tensorflow/transformer/attention.py start line: 0 end line: 0 size: 50 LOC McCabe index: 7 number of parameters: 17 id: 66 unit: def get_final_text() file: mesh_tensorflow/bert/run_squad.py start line: 0 end line: 0 size: 50 LOC McCabe index: 21 number of parameters: 3 id: 67 unit: def batch_norm() file: mesh_tensorflow/layers.py start line: 0 end line: 0 size: 49 LOC McCabe index: 8 number of parameters: 8 id: 68 unit: def gating() file: mesh_tensorflow/transformer/universal_transformer.py start line: 0 end line: 0 size: 49 LOC McCabe index: 3 number of parameters: 5 id: 69 unit: def call() file: mesh_tensorflow/transformer/transformer_layers.py start line: 0 end line: 0 size: 49 LOC McCabe index: 8 number of parameters: 4 id: 70 unit: def init_weights() file: mesh_tensorflow/transformer/fixup_layers.py start line: 0 end line: 0 size: 49 LOC McCabe index: 5 number of parameters: 1 id: 71 unit: def init_weights() file: mesh_tensorflow/transformer/attention.py start line: 0 end line: 0 size: 49 LOC McCabe index: 7 number of parameters: 1 id: 72 unit: def decode_from_dataset() file: mesh_tensorflow/transformer/utils.py start line: 0 end line: 0 size: 47 LOC McCabe index: 6 number of parameters: 9 id: 73 unit: def record() file: mesh_tensorflow/experimental/unet.py start line: 0 end line: 0 size: 46 LOC McCabe index: 9 number of parameters: 3 id: 74 unit: def score_from_strings() file: mesh_tensorflow/transformer/utils.py start line: 0 end line: 0 size: 46 LOC McCabe index: 17 number of parameters: 13 id: 75 unit: def talking_heads() file: mesh_tensorflow/transformer/transformer_layers.py start line: 0 end line: 0 size: 46 LOC McCabe index: 16 number of parameters: 7 id: 76 unit: def attention() file: mesh_tensorflow/transformer/attention.py start line: 0 end line: 0 size: 46 LOC McCabe index: 8 number of parameters: 13 id: 77 unit: def export_model() file: mesh_tensorflow/transformer/utils.py start line: 0 end line: 0 size: 45 LOC McCabe index: 11 number of parameters: 8 id: 78 unit: def call() file: mesh_tensorflow/transformer/transformer_layers.py start line: 0 end line: 0 size: 45 LOC McCabe index: 9 number of parameters: 4 id: 79 unit: def input_fn_builder() file: mesh_tensorflow/bert/run_pretraining.py start line: 0 end line: 0 size: 45 LOC McCabe index: 5 number of parameters: 5 id: 80 unit: def local_2d_self_attention_spatial_blocks() file: mesh_tensorflow/layers.py start line: 0 end line: 0 size: 44 LOC McCabe index: 9 number of parameters: 10 id: 81 unit: def multihead_self_attention_memory_compressed() file: mesh_tensorflow/layers.py start line: 0 end line: 0 size: 44 LOC McCabe index: 2 number of parameters: 11 id: 82 unit: def alltoall_ring() file: mesh_tensorflow/placement_mesh_impl.py start line: 0 end line: 0 size: 44 LOC McCabe index: 10 number of parameters: 4 id: 83 unit: def flexible_ckpt_init_mapping() file: mesh_tensorflow/transformer/utils.py start line: 0 end line: 0 size: 44 LOC McCabe index: 15 number of parameters: 3 id: 84 unit: def __init__() file: mesh_tensorflow/transformer/vocab_embeddings.py start line: 0 end line: 0 size: 44 LOC McCabe index: 5 number of parameters: 8 id: 85 unit: def maybe_reshape_attention_input_for_2d_sharding() file: mesh_tensorflow/transformer/attention.py start line: 0 end line: 0 size: 43 LOC McCabe index: 20 number of parameters: 6 id: 86 unit: def self_attention() file: mesh_tensorflow/bert/bert.py start line: 0 end line: 0 size: 43 LOC McCabe index: 3 number of parameters: 3 id: 87 unit: def packed_parallel_tsv_dataset() file: mesh_tensorflow/transformer/dataset.py start line: 0 end line: 0 size: 42 LOC McCabe index: 6 number of parameters: 8 id: 88 unit: def ut_highway() file: mesh_tensorflow/transformer/universal_transformer.py start line: 0 end line: 0 size: 42 LOC McCabe index: 7 number of parameters: 4 id: 89 unit: def multihead_attention() file: mesh_tensorflow/layers.py start line: 0 end line: 0 size: 41 LOC McCabe index: 4 number of parameters: 11 id: 90 unit: def __init__() file: mesh_tensorflow/placement_mesh_impl.py start line: 0 end line: 0 size: 41 LOC McCabe index: 5 number of parameters: 3 id: 91 unit: def ffn_layer_multi_inputs() file: mesh_tensorflow/transformer/universal_transformer.py start line: 0 end line: 0 size: 40 LOC McCabe index: 9 number of parameters: 9 id: 92 unit: def attention_params() file: mesh_tensorflow/transformer/transformer_layers.py start line: 0 end line: 0 size: 40 LOC McCabe index: 5 number of parameters: 9 id: 93 unit: def __init__() file: mesh_tensorflow/transformer/fixup_layers.py start line: 0 end line: 0 size: 40 LOC McCabe index: 3 number of parameters: 16 id: 94 unit: def __init__() file: mesh_tensorflow/transformer/transformer_layers.py start line: 0 end line: 0 size: 39 LOC McCabe index: 4 number of parameters: 15 id: 95 unit: def local_self_attention_spatial_blocks() file: mesh_tensorflow/layers.py start line: 0 end line: 0 size: 38 LOC McCabe index: 9 number of parameters: 9 id: 96 unit: def __init__() file: mesh_tensorflow/transformer/transformer_layers.py start line: 0 end line: 0 size: 38 LOC McCabe index: 1 number of parameters: 19 id: 97 unit: def enc_dec_attention() file: mesh_tensorflow/transformer/transformer_layers.py start line: 0 end line: 0 size: 38 LOC McCabe index: 7 number of parameters: 7 id: 98 unit: def __init__() file: mesh_tensorflow/transformer/funnel_transformer.py start line: 0 end line: 0 size: 38 LOC McCabe index: 7 number of parameters: 11 id: 99 unit: def multihead_self_attention_incremental() file: mesh_tensorflow/layers.py start line: 0 end line: 0 size: 37 LOC McCabe index: 1 number of parameters: 7 id: 100 unit: def print_solution() file: mesh_tensorflow/auto_mtf/print_cp_model_solution.py start line: 0 end line: 0 size: 37 LOC McCabe index: 15 number of parameters: 2 id: 101 unit: def make_params() file: mesh_tensorflow/transformer/transformer_layers.py start line: 0 end line: 0 size: 37 LOC McCabe index: 5 number of parameters: 2 id: 102 unit: def _get_decoder_inputs() file: mesh_tensorflow/transformer/transformer_layers.py start line: 0 end line: 0 size: 37 LOC McCabe index: 5 number of parameters: 2 id: 103 unit: def hidden_to_logits() file: mesh_tensorflow/transformer/vocab_embeddings.py start line: 0 end line: 0 size: 37 LOC McCabe index: 3 number of parameters: 3 id: 104 unit: def input_fn_builder() file: mesh_tensorflow/bert/run_classifier.py start line: 0 end line: 0 size: 37 LOC McCabe index: 5 number of parameters: 4 id: 105 unit: def __init__() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 36 LOC McCabe index: 5 number of parameters: 5 id: 106 unit: def _create_cluster() file: mesh_tensorflow/transformer/adaptive_softmax.py start line: 0 end line: 0 size: 36 LOC McCabe index: 6 number of parameters: 4 id: 107 unit: def enc_dec_attention_bias() file: mesh_tensorflow/transformer/transformer_layers.py start line: 0 end line: 0 size: 36 LOC McCabe index: 11 number of parameters: 4 id: 108 unit: def call() file: mesh_tensorflow/transformer/transformer_layers.py start line: 0 end line: 0 size: 36 LOC McCabe index: 5 number of parameters: 4 id: 109 unit: def projective_transform() file: mesh_tensorflow/experimental/data_aug_lib.py start line: 0 end line: 0 size: 35 LOC McCabe index: 8 number of parameters: 6 id: 110 unit: def _pack_with_custom_ops() file: mesh_tensorflow/transformer/dataset.py start line: 0 end line: 0 size: 35 LOC McCabe index: 7 number of parameters: 3 id: 111 unit: def call() file: mesh_tensorflow/transformer/moe.py start line: 0 end line: 0 size: 35 LOC McCabe index: 8 number of parameters: 4 id: 112 unit: def score_with_estimator_lazy() file: mesh_tensorflow/transformer/utils.py start line: 0 end line: 0 size: 35 LOC McCabe index: 10 number of parameters: 8 id: 113 unit: def __init__() file: mesh_tensorflow/transformer/vocab_embeddings.py start line: 0 end line: 0 size: 35 LOC McCabe index: 2 number of parameters: 8 id: 114 unit: def call() file: mesh_tensorflow/transformer/memory_layers.py start line: 0 end line: 0 size: 35 LOC McCabe index: 3 number of parameters: 3 id: 115 unit: def make_params() file: mesh_tensorflow/transformer/fixup_layers.py start line: 0 end line: 0 size: 35 LOC McCabe index: 5 number of parameters: 2 id: 116 unit: def greedy_decode() file: mesh_tensorflow/beam_search.py start line: 0 end line: 0 size: 34 LOC McCabe index: 3 number of parameters: 7 id: 117 unit: def slicewise() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 34 LOC McCabe index: 8 number of parameters: 7 id: 118 unit: def _tf_upscale() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 34 LOC McCabe index: 4 number of parameters: 4 id: 119 unit: def _shared_params() file: mesh_tensorflow/transformer/transformer.py start line: 0 end line: 0 size: 34 LOC McCabe index: 9 number of parameters: 3 id: 120 unit: def compute_loss() file: mesh_tensorflow/transformer/adaptive_softmax.py start line: 0 end line: 0 size: 34 LOC McCabe index: 4 number of parameters: 5 id: 121 unit: def validate_case_matches_checkpoint() file: mesh_tensorflow/bert/tokenization.py start line: 0 end line: 0 size: 34 LOC McCabe index: 9 number of parameters: 2 id: 122 unit: def allreduce_ring_single_shard() file: mesh_tensorflow/placement_mesh_impl.py start line: 0 end line: 0 size: 33 LOC McCabe index: 8 number of parameters: 3 id: 123 unit: def _initialize_splittable_and_unsplittable_dims() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 33 LOC McCabe index: 7 number of parameters: 3 id: 124 unit: def hybrid_attention() file: mesh_tensorflow/transformer/attention.py start line: 0 end line: 0 size: 33 LOC McCabe index: 2 number of parameters: 11 id: 125 unit: def create_model() file: mesh_tensorflow/bert/run_classifier.py start line: 0 end line: 0 size: 33 LOC McCabe index: 1 number of parameters: 9 id: 126 unit: def __init__() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 32 LOC McCabe index: 4 number of parameters: 8 id: 127 unit: def _reduce_helper() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 32 LOC McCabe index: 10 number of parameters: 4 id: 128 unit: def __init__() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 32 LOC McCabe index: 7 number of parameters: 6 id: 129 unit: def call() file: mesh_tensorflow/transformer/heterogeneous_moe.py start line: 0 end line: 0 size: 32 LOC McCabe index: 6 number of parameters: 4 id: 130 unit: def call() file: mesh_tensorflow/transformer/funnel_transformer.py start line: 0 end line: 0 size: 32 LOC McCabe index: 13 number of parameters: 3 id: 131 unit: def tokenize() file: mesh_tensorflow/bert/tokenization.py start line: 0 end line: 0 size: 32 LOC McCabe index: 9 number of parameters: 2 id: 132 unit: def __init__() file: mesh_tensorflow/experimental/input_reader.py start line: 0 end line: 0 size: 31 LOC McCabe index: 2 number of parameters: 2 id: 133 unit: def lower() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 31 LOC McCabe index: 6 number of parameters: 2 id: 134 unit: def generate_heterogeneous_expert_masks() file: mesh_tensorflow/transformer/heterogeneous_moe.py start line: 0 end line: 0 size: 31 LOC McCabe index: 9 number of parameters: 5 id: 135 unit: def input_fn_builder() file: mesh_tensorflow/bert/run_squad.py start line: 0 end line: 0 size: 31 LOC McCabe index: 7 number of parameters: 4 id: 136 unit: def __call__() file: mesh_tensorflow/layers.py start line: 0 end line: 0 size: 30 LOC McCabe index: 11 number of parameters: 3 id: 137 unit: def call() file: mesh_tensorflow/transformer/transformer_layers.py start line: 0 end line: 0 size: 30 LOC McCabe index: 5 number of parameters: 5 id: 138 unit: def call() file: mesh_tensorflow/transformer/fixup_layers.py start line: 0 end line: 0 size: 30 LOC McCabe index: 3 number of parameters: 4 id: 139 unit: def file_based_input_fn_builder() file: mesh_tensorflow/bert/run_classifier.py start line: 0 end line: 0 size: 30 LOC McCabe index: 6 number of parameters: 4 id: 140 unit: def separable_conv1d() file: mesh_tensorflow/layers.py start line: 0 end line: 0 size: 29 LOC McCabe index: 1 number of parameters: 9 id: 141 unit: def conv3d_transpose_with_blocks() file: mesh_tensorflow/layers.py start line: 0 end line: 0 size: 29 LOC McCabe index: 10 number of parameters: 5 id: 142 unit: def masked_local_attention_1d_incremental() file: mesh_tensorflow/layers.py start line: 0 end line: 0 size: 29 LOC McCabe index: 2 number of parameters: 8 id: 143 unit: def allreduce() file: mesh_tensorflow/simd_mesh_impl.py start line: 0 end line: 0 size: 29 LOC McCabe index: 8 number of parameters: 4 id: 144 unit: def allreduce_ring() file: mesh_tensorflow/placement_mesh_impl.py start line: 0 end line: 0 size: 29 LOC McCabe index: 8 number of parameters: 3 id: 145 unit: def pretokenized_tfds_dataset() file: mesh_tensorflow/transformer/dataset.py start line: 0 end line: 0 size: 29 LOC McCabe index: 3 number of parameters: 7 id: 146 unit: def __init__() file: mesh_tensorflow/transformer/adaptive_softmax.py start line: 0 end line: 0 size: 29 LOC McCabe index: 5 number of parameters: 10 id: 147 unit: def get_timing_signal_1d() file: mesh_tensorflow/transformer/universal_transformer.py start line: 0 end line: 0 size: 29 LOC McCabe index: 2 number of parameters: 7 id: 148 unit: def _build_ckpt_to_local_var_name_mapping() file: mesh_tensorflow/transformer/utils.py start line: 0 end line: 0 size: 29 LOC McCabe index: 13 number of parameters: 6 id: 149 unit: def save_scores_to_tfrecords() file: mesh_tensorflow/transformer/utils.py start line: 0 end line: 0 size: 29 LOC McCabe index: 7 number of parameters: 5 id: 150 unit: def call() file: mesh_tensorflow/transformer/transformer_layers.py start line: 0 end line: 0 size: 29 LOC McCabe index: 5 number of parameters: 4 id: 151 unit: def __init__() file: mesh_tensorflow/transformer/vocab_embeddings.py start line: 0 end line: 0 size: 29 LOC McCabe index: 1 number of parameters: 8 id: 152 unit: def conv3d_with_blocks() file: mesh_tensorflow/layers.py start line: 0 end line: 0 size: 28 LOC McCabe index: 10 number of parameters: 5 id: 153 unit: def __init__() file: mesh_tensorflow/simd_mesh_impl.py start line: 0 end line: 0 size: 28 LOC McCabe index: 5 number of parameters: 7 id: 154 unit: def compute_cost_graph() file: mesh_tensorflow/auto_mtf/graph_interface.py start line: 0 end line: 0 size: 28 LOC McCabe index: 9 number of parameters: 2 id: 155 unit: def conv_with_spatial_partition() file: mesh_tensorflow/experimental/unet.py start line: 0 end line: 0 size: 28 LOC McCabe index: 3 number of parameters: 11 id: 156 unit: def __init__() file: mesh_tensorflow/optimize.py start line: 0 end line: 0 size: 28 LOC McCabe index: 5 number of parameters: 12 id: 157 unit: def __init__() file: mesh_tensorflow/transformer/moe.py start line: 0 end line: 0 size: 28 LOC McCabe index: 1 number of parameters: 14 id: 158 unit: def decode_from_file() file: mesh_tensorflow/transformer/utils.py start line: 0 end line: 0 size: 28 LOC McCabe index: 1 number of parameters: 10 id: 159 unit: def save_scores() file: mesh_tensorflow/transformer/utils.py start line: 0 end line: 0 size: 28 LOC McCabe index: 15 number of parameters: 4 id: 160 unit: def train_model() file: mesh_tensorflow/transformer/utils.py start line: 0 end line: 0 size: 28 LOC McCabe index: 8 number of parameters: 11 id: 161 unit: def relative_position_spans() file: mesh_tensorflow/transformer/transformer_layers.py start line: 0 end line: 0 size: 28 LOC McCabe index: 3 number of parameters: 2 id: 162 unit: def get_masked_lm_output() file: mesh_tensorflow/bert/bert.py start line: 0 end line: 0 size: 28 LOC McCabe index: 5 number of parameters: 4 id: 163 unit: def _depthwise_conv1d_hack() file: mesh_tensorflow/layers.py start line: 0 end line: 0 size: 27 LOC McCabe index: 4 number of parameters: 9 id: 164 unit: def pretokenized_t2t_dataset() file: mesh_tensorflow/transformer/dataset.py start line: 0 end line: 0 size: 27 LOC McCabe index: 3 number of parameters: 9 id: 165 unit: def pack_dataset() file: mesh_tensorflow/transformer/dataset.py start line: 0 end line: 0 size: 27 LOC McCabe index: 14 number of parameters: 4 id: 166 unit: def make_bitransformer() file: mesh_tensorflow/transformer/transformer.py start line: 0 end line: 0 size: 27 LOC McCabe index: 1 number of parameters: 7 id: 167 unit: def call() file: mesh_tensorflow/transformer/universal_transformer.py start line: 0 end line: 0 size: 27 LOC McCabe index: 11 number of parameters: 3 id: 168 unit: def build_model() file: mesh_tensorflow/transformer/utils.py start line: 0 end line: 0 size: 27 LOC McCabe index: 4 number of parameters: 5 id: 169 unit: def serialize_num_microbatches() file: mesh_tensorflow/transformer/utils.py start line: 0 end line: 0 size: 27 LOC McCabe index: 4 number of parameters: 5 id: 170 unit: def __init__() file: mesh_tensorflow/transformer/transformer_layers.py start line: 0 end line: 0 size: 27 LOC McCabe index: 2 number of parameters: 14 id: 171 unit: def __init__() file: mesh_tensorflow/transformer/evolved_transformer.py start line: 0 end line: 0 size: 27 LOC McCabe index: 1 number of parameters: 5 id: 172 unit: def conv2d_transpose_with_blocks() file: mesh_tensorflow/layers.py start line: 0 end line: 0 size: 26 LOC McCabe index: 10 number of parameters: 4 id: 173 unit: def multihead_attention_params() file: mesh_tensorflow/layers.py start line: 0 end line: 0 size: 26 LOC McCabe index: 5 number of parameters: 6 id: 174 unit: def annealed_dropout() file: mesh_tensorflow/layers.py start line: 0 end line: 0 size: 26 LOC McCabe index: 6 number of parameters: 7 id: 175 unit: def evaluate_layout() file: mesh_tensorflow/auto_mtf/layout_optimizer.py start line: 0 end line: 0 size: 26 LOC McCabe index: 8 number of parameters: 2 id: 176 unit: def initialize() file: mesh_tensorflow/transformer/transformer.py start line: 0 end line: 0 size: 26 LOC McCabe index: 10 number of parameters: 1 id: 177 unit: def __init__() file: mesh_tensorflow/transformer/fixup_layers.py start line: 0 end line: 0 size: 26 LOC McCabe index: 3 number of parameters: 8 id: 178 unit: def _compute_merge_qkv() file: mesh_tensorflow/transformer/attention.py start line: 0 end line: 0 size: 26 LOC McCabe index: 5 number of parameters: 2 id: 179 unit: def moe() file: mesh_tensorflow/bert/bert.py start line: 0 end line: 0 size: 26 LOC McCabe index: 2 number of parameters: 6 id: 180 unit: def get_dense_kernel_weights() file: mesh_tensorflow/layers.py start line: 0 end line: 0 size: 25 LOC McCabe index: 4 number of parameters: 9 id: 181 unit: def conv2d_with_blocks() file: mesh_tensorflow/layers.py start line: 0 end line: 0 size: 25 LOC McCabe index: 10 number of parameters: 4 id: 182 unit: def _reversible_half_residual_grad() file: mesh_tensorflow/layers.py start line: 0 end line: 0 size: 25 LOC McCabe index: 4 number of parameters: 5 id: 183 unit: def __init__() file: mesh_tensorflow/experimental/model_executor.py start line: 0 end line: 0 size: 25 LOC McCabe index: 3 number of parameters: 5 id: 184 unit: def __init__() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 25 LOC McCabe index: 5 number of parameters: 4 id: 185 unit: def pretokenized_tfrecord_dataset() file: mesh_tensorflow/transformer/dataset.py start line: 0 end line: 0 size: 25 LOC McCabe index: 9 number of parameters: 7 id: 186 unit: def make_layer_stack() file: mesh_tensorflow/transformer/transformer.py start line: 0 end line: 0 size: 25 LOC McCabe index: 11 number of parameters: 4 id: 187 unit: def add_position_timing_signal_func() file: mesh_tensorflow/transformer/universal_transformer.py start line: 0 end line: 0 size: 25 LOC McCabe index: 8 number of parameters: 4 id: 188 unit: def call() file: mesh_tensorflow/transformer/moe.py start line: 0 end line: 0 size: 25 LOC McCabe index: 5 number of parameters: 4 id: 189 unit: def compute_batch_size() file: mesh_tensorflow/transformer/utils.py start line: 0 end line: 0 size: 25 LOC McCabe index: 6 number of parameters: 4 id: 190 unit: def call() file: mesh_tensorflow/transformer/transformer_layers.py start line: 0 end line: 0 size: 25 LOC McCabe index: 5 number of parameters: 4 id: 191 unit: def call() file: mesh_tensorflow/transformer/evolved_transformer.py start line: 0 end line: 0 size: 25 LOC McCabe index: 1 number of parameters: 4 id: 192 unit: def layer_norm() file: mesh_tensorflow/bert/bert.py start line: 0 end line: 0 size: 25 LOC McCabe index: 4 number of parameters: 7 id: 193 unit: def __init__() file: mesh_tensorflow/bert/run_squad.py start line: 0 end line: 0 size: 25 LOC McCabe index: 1 number of parameters: 13 id: 194 unit: def _get_hosts_to_hold_ds() file: mesh_tensorflow/experimental/input_reader.py start line: 0 end line: 0 size: 24 LOC McCabe index: 4 number of parameters: 2 id: 195 unit: def create_host_call() file: mesh_tensorflow/utils.py start line: 0 end line: 0 size: 24 LOC McCabe index: 8 number of parameters: 1 id: 196 unit: def clone_operations() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 24 LOC McCabe index: 8 number of parameters: 3 id: 197 unit: def combine_slices() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 24 LOC McCabe index: 8 number of parameters: 4 id: 198 unit: def apply_grad() file: mesh_tensorflow/optimize.py start line: 0 end line: 0 size: 24 LOC McCabe index: 4 number of parameters: 3 id: 199 unit: def __init__() file: mesh_tensorflow/transformer/universal_transformer.py start line: 0 end line: 0 size: 24 LOC McCabe index: 1 number of parameters: 0 id: 200 unit: def tpu_mesh_shape() file: mesh_tensorflow/transformer/utils.py start line: 0 end line: 0 size: 24 LOC McCabe index: 10 number of parameters: 3 id: 201 unit: def encode_inputs() file: mesh_tensorflow/transformer/utils.py start line: 0 end line: 0 size: 24 LOC McCabe index: 9 number of parameters: 7 id: 202 unit: def _maybe_add_pretokenized_features() file: mesh_tensorflow/transformer/utils.py start line: 0 end line: 0 size: 24 LOC McCabe index: 12 number of parameters: 2 id: 203 unit: def get_indices() file: mesh_tensorflow/transformer/memory_layers.py start line: 0 end line: 0 size: 24 LOC McCabe index: 2 number of parameters: 3 id: 204 unit: def dense_product() file: mesh_tensorflow/layers.py start line: 0 end line: 0 size: 23 LOC McCabe index: 6 number of parameters: 6 id: 205 unit: def solve() file: mesh_tensorflow/auto_mtf/layout_optimizer.py start line: 0 end line: 0 size: 23 LOC McCabe index: 8 number of parameters: 2 id: 206 unit: def _list_cpu_gpu_devices() file: mesh_tensorflow/experimental/model_executor.py start line: 0 end line: 0 size: 23 LOC McCabe index: 9 number of parameters: 2 id: 207 unit: def get_input_mtf_shapes() file: mesh_tensorflow/experimental/unet.py start line: 0 end line: 0 size: 23 LOC McCabe index: 3 number of parameters: 1 id: 208 unit: def maybe_rot180() file: mesh_tensorflow/experimental/data_aug_lib.py start line: 0 end line: 0 size: 23 LOC McCabe index: 2 number of parameters: 4 id: 209 unit: def _stochastically_use_non_top_expert() file: mesh_tensorflow/transformer/moe.py start line: 0 end line: 0 size: 23 LOC McCabe index: 2 number of parameters: 3 id: 210 unit: def _compute_num_blocks_and_layer() file: mesh_tensorflow/transformer/utils.py start line: 0 end line: 0 size: 23 LOC McCabe index: 7 number of parameters: 1 id: 211 unit: def encode_delimited_lm() file: mesh_tensorflow/transformer/utils.py start line: 0 end line: 0 size: 23 LOC McCabe index: 4 number of parameters: 7 id: 212 unit: def __init__() file: mesh_tensorflow/transformer/transformer_layers.py start line: 0 end line: 0 size: 23 LOC McCabe index: 5 number of parameters: 3 id: 213 unit: def layer_output_from_attention_output() file: mesh_tensorflow/transformer/transformer_layers.py start line: 0 end line: 0 size: 23 LOC McCabe index: 1 number of parameters: 4 id: 214 unit: def get_assignment_map_from_checkpoint() file: mesh_tensorflow/bert/bert.py start line: 0 end line: 0 size: 23 LOC McCabe index: 8 number of parameters: 2 id: 215 unit: def file_based_convert_examples_to_features() file: mesh_tensorflow/bert/run_classifier.py start line: 0 end line: 0 size: 23 LOC McCabe index: 3 number of parameters: 5 id: 216 unit: def softmax_cross_entropy_with_logits() file: mesh_tensorflow/layers.py start line: 0 end line: 0 size: 22 LOC McCabe index: 6 number of parameters: 4 id: 217 unit: def clip_activation_gradient() file: mesh_tensorflow/layers.py start line: 0 end line: 0 size: 22 LOC McCabe index: 3 number of parameters: 2 id: 218 unit: def _dataset_creator() file: mesh_tensorflow/experimental/offline_data_aug.py start line: 0 end line: 0 size: 22 LOC McCabe index: 1 number of parameters: 0 id: 219 unit: def deconv_with_spatial_partition() file: mesh_tensorflow/experimental/unet.py start line: 0 end line: 0 size: 22 LOC McCabe index: 2 number of parameters: 10 id: 220 unit: def maybe_flip() file: mesh_tensorflow/experimental/data_aug_lib.py start line: 0 end line: 0 size: 22 LOC McCabe index: 2 number of parameters: 4 id: 221 unit: def _compute_loss() file: mesh_tensorflow/transformer/transformer.py start line: 0 end line: 0 size: 22 LOC McCabe index: 5 number of parameters: 5 id: 222 unit: def hidden_to_logits() file: mesh_tensorflow/transformer/vocab_embeddings.py start line: 0 end line: 0 size: 22 LOC McCabe index: 1 number of parameters: 3 id: 223 unit: def validate_flags_or_throw() file: mesh_tensorflow/bert/run_squad.py start line: 0 end line: 0 size: 22 LOC McCabe index: 9 number of parameters: 1 id: 224 unit: def local_2d_halo_exchange() file: mesh_tensorflow/layers.py start line: 0 end line: 0 size: 21 LOC McCabe index: 6 number of parameters: 7 id: 225 unit: def dot_product_attention() file: mesh_tensorflow/layers.py start line: 0 end line: 0 size: 21 LOC McCabe index: 2 number of parameters: 8 id: 226 unit: def _gen_copy_master_to_slices_op() file: mesh_tensorflow/simd_mesh_impl.py start line: 0 end line: 0 size: 21 LOC McCabe index: 8 number of parameters: 5 id: 227 unit: def receive() file: mesh_tensorflow/simd_mesh_impl.py start line: 0 end line: 0 size: 21 LOC McCabe index: 6 number of parameters: 4 id: 228 unit: def _initialize_variables() file: mesh_tensorflow/auto_mtf/layout_optimizer.py start line: 0 end line: 0 size: 21 LOC McCabe index: 7 number of parameters: 1 id: 229 unit: def _get_pnum_map() file: mesh_tensorflow/experimental/input_reader.py start line: 0 end line: 0 size: 21 LOC McCabe index: 6 number of parameters: 2 id: 230 unit: def _gen_rand_mask() file: mesh_tensorflow/experimental/data_aug_lib.py start line: 0 end line: 0 size: 21 LOC McCabe index: 2 number of parameters: 5 id: 231 unit: def lower() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 21 LOC McCabe index: 6 number of parameters: 2 id: 232 unit: def binary_arguments_to_tensors() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 21 LOC McCabe index: 6 number of parameters: 2 id: 233 unit: def sublayer_annealed_dropout() file: mesh_tensorflow/transformer/transformer.py start line: 0 end line: 0 size: 21 LOC McCabe index: 5 number of parameters: 6 id: 234 unit: def sinusoid_positional_embedding_weights() file: mesh_tensorflow/transformer/transformer.py start line: 0 end line: 0 size: 21 LOC McCabe index: 2 number of parameters: 6 id: 235 unit: def add_step_timing_signal_func() file: mesh_tensorflow/transformer/universal_transformer.py start line: 0 end line: 0 size: 21 LOC McCabe index: 6 number of parameters: 4 id: 236 unit: def call() file: mesh_tensorflow/transformer/transformer_layers.py start line: 0 end line: 0 size: 21 LOC McCabe index: 3 number of parameters: 4 id: 237 unit: def attention_internal() file: mesh_tensorflow/transformer/transformer_layers.py start line: 0 end line: 0 size: 21 LOC McCabe index: 6 number of parameters: 9 id: 238 unit: def call() file: mesh_tensorflow/transformer/transformer_layers.py start line: 0 end line: 0 size: 21 LOC McCabe index: 5 number of parameters: 4 id: 239 unit: def __init__() file: mesh_tensorflow/transformer/transformer_layers.py start line: 0 end line: 0 size: 21 LOC McCabe index: 3 number of parameters: 9 id: 240 unit: def create_model() file: mesh_tensorflow/bert/run_squad.py start line: 0 end line: 0 size: 21 LOC McCabe index: 1 number of parameters: 5 id: 241 unit: def _preprocess_input() file: mesh_tensorflow/auto_mtf/layout_optimizer.py start line: 0 end line: 0 size: 20 LOC McCabe index: 5 number of parameters: 1 id: 242 unit: def train_and_eval() file: mesh_tensorflow/experimental/model_executor.py start line: 0 end line: 0 size: 20 LOC McCabe index: 4 number of parameters: 0 id: 243 unit: def __init__() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 20 LOC McCabe index: 8 number of parameters: 4 id: 244 unit: def _add_token_emb_to_gate_inputs() file: mesh_tensorflow/transformer/moe.py start line: 0 end line: 0 size: 20 LOC McCabe index: 6 number of parameters: 3 id: 245 unit: def _relative_position_bucket() file: mesh_tensorflow/transformer/transformer_layers.py start line: 0 end line: 0 size: 20 LOC McCabe index: 2 number of parameters: 4 id: 246 unit: def get_kernel_wt() file: mesh_tensorflow/transformer/transformer_layers.py start line: 0 end line: 0 size: 20 LOC McCabe index: 1 number of parameters: 9 id: 247 unit: def get_kernel_wt() file: mesh_tensorflow/transformer/transformer_layers.py start line: 0 end line: 0 size: 20 LOC McCabe index: 1 number of parameters: 9 id: 248 unit: def main() file: mesh_tensorflow/transformer/main.py start line: 0 end line: 0 size: 20 LOC McCabe index: 4 number of parameters: 1 id: 249 unit: def _relative_position_bucket() file: mesh_tensorflow/bert/bert.py start line: 0 end line: 0 size: 20 LOC McCabe index: 2 number of parameters: 4 id: 250 unit: def process_feature() file: mesh_tensorflow/bert/run_squad.py start line: 0 end line: 0 size: 20 LOC McCabe index: 3 number of parameters: 2 id: 251 unit: def layer_norm() file: mesh_tensorflow/layers.py start line: 0 end line: 0 size: 19 LOC McCabe index: 1 number of parameters: 4 id: 252 unit: def layout_and_mesh_shape() file: mesh_tensorflow/auto_mtf/api.py start line: 0 end line: 0 size: 19 LOC McCabe index: 5 number of parameters: 3 id: 253 unit: def _mesh_shape_iterator() file: mesh_tensorflow/auto_mtf/api.py start line: 0 end line: 0 size: 19 LOC McCabe index: 8 number of parameters: 2 id: 254 unit: def create_graph_mesh_and_mesh_impl() file: mesh_tensorflow/experimental/model_executor.py start line: 0 end line: 0 size: 19 LOC McCabe index: 2 number of parameters: 1 id: 255 unit: def lower() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 19 LOC McCabe index: 4 number of parameters: 2 id: 256 unit: def lower() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 19 LOC McCabe index: 3 number of parameters: 2 id: 257 unit: def untokenized_tfds_dataset() file: mesh_tensorflow/transformer/dataset.py start line: 0 end line: 0 size: 19 LOC McCabe index: 3 number of parameters: 8 id: 258 unit: def __init__() file: mesh_tensorflow/transformer/transformer.py start line: 0 end line: 0 size: 19 LOC McCabe index: 3 number of parameters: 8 id: 259 unit: def call() file: mesh_tensorflow/transformer/transformer.py start line: 0 end line: 0 size: 19 LOC McCabe index: 4 number of parameters: 3 id: 260 unit: def __init__() file: mesh_tensorflow/transformer/transformer.py start line: 0 end line: 0 size: 19 LOC McCabe index: 2 number of parameters: 8 id: 261 unit: def decode() file: mesh_tensorflow/transformer/utils.py start line: 0 end line: 0 size: 19 LOC McCabe index: 5 number of parameters: 4 id: 262 unit: def infer_model() file: mesh_tensorflow/transformer/utils.py start line: 0 end line: 0 size: 19 LOC McCabe index: 6 number of parameters: 9 id: 263 unit: def call() file: mesh_tensorflow/transformer/transformer_layers.py start line: 0 end line: 0 size: 19 LOC McCabe index: 6 number of parameters: 4 id: 264 unit: def attention_bias_local_2d_block() file: mesh_tensorflow/layers.py start line: 0 end line: 0 size: 18 LOC McCabe index: 1 number of parameters: 6 id: 265 unit: def __init__() file: mesh_tensorflow/experimental/input_reader.py start line: 0 end line: 0 size: 18 LOC McCabe index: 2 number of parameters: 7 id: 266 unit: def combine_assignments() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 18 LOC McCabe index: 7 number of parameters: 2 id: 267 unit: def compute_lr_for_step() file: mesh_tensorflow/optimize.py start line: 0 end line: 0 size: 18 LOC McCabe index: 4 number of parameters: 3 id: 268 unit: def call() file: mesh_tensorflow/transformer/transformer.py start line: 0 end line: 0 size: 18 LOC McCabe index: 5 number of parameters: 3 id: 269 unit: def sublayer_rms_norm_subsampled() file: mesh_tensorflow/transformer/transformer.py start line: 0 end line: 0 size: 18 LOC McCabe index: 1 number of parameters: 5 id: 270 unit: def call() file: mesh_tensorflow/transformer/transformer_layers.py start line: 0 end line: 0 size: 18 LOC McCabe index: 6 number of parameters: 4 id: 271 unit: def __init__() file: mesh_tensorflow/transformer/transformer_layers.py start line: 0 end line: 0 size: 18 LOC McCabe index: 1 number of parameters: 8 id: 272 unit: def conv3d_transpose() file: mesh_tensorflow/layers.py start line: 0 end line: 0 size: 17 LOC McCabe index: 3 number of parameters: 5 id: 273 unit: def dense_relu_dense() file: mesh_tensorflow/layers.py start line: 0 end line: 0 size: 17 LOC McCabe index: 1 number of parameters: 8 id: 274 unit: def embedding_weights() file: mesh_tensorflow/layers.py start line: 0 end line: 0 size: 17 LOC McCabe index: 4 number of parameters: 7 id: 275 unit: def reversible_half_residual_and_swap() file: mesh_tensorflow/layers.py start line: 0 end line: 0 size: 17 LOC McCabe index: 4 number of parameters: 6 id: 276 unit: def compute_memory_contents_under_schedule() file: mesh_tensorflow/auto_mtf/graph_interface.py start line: 0 end line: 0 size: 17 LOC McCabe index: 7 number of parameters: 2 id: 277 unit: def __init__() file: mesh_tensorflow/experimental/input_reader.py start line: 0 end line: 0 size: 17 LOC McCabe index: 2 number of parameters: 6 id: 278 unit: def maybe_add_noise() file: mesh_tensorflow/experimental/data_aug_lib.py start line: 0 end line: 0 size: 17 LOC McCabe index: 4 number of parameters: 6 id: 279 unit: def allconcat_ring() file: mesh_tensorflow/placement_mesh_impl.py start line: 0 end line: 0 size: 17 LOC McCabe index: 9 number of parameters: 3 id: 280 unit: def unique_name() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 17 LOC McCabe index: 6 number of parameters: 3 id: 281 unit: def adafactor_optimizer_from_hparams() file: mesh_tensorflow/optimize.py start line: 0 end line: 0 size: 17 LOC McCabe index: 3 number of parameters: 2 id: 282 unit: def apply_grad() file: mesh_tensorflow/optimize.py start line: 0 end line: 0 size: 17 LOC McCabe index: 6 number of parameters: 3 id: 283 unit: def apply_grad() file: mesh_tensorflow/optimize.py start line: 0 end line: 0 size: 17 LOC McCabe index: 6 number of parameters: 3 id: 284 unit: def make_bi_student_teacher() file: mesh_tensorflow/transformer/transformer.py start line: 0 end line: 0 size: 17 LOC McCabe index: 1 number of parameters: 4 id: 285 unit: def vanilla_transformer_layer() file: mesh_tensorflow/transformer/universal_transformer.py start line: 0 end line: 0 size: 17 LOC McCabe index: 7 number of parameters: 4 id: 286 unit: def _get_combined_dataset_input_fn() file: mesh_tensorflow/transformer/utils.py start line: 0 end line: 0 size: 17 LOC McCabe index: 5 number of parameters: 4 id: 287 unit: def call() file: mesh_tensorflow/transformer/transformer_layers.py start line: 0 end line: 0 size: 17 LOC McCabe index: 1 number of parameters: 4 id: 288 unit: def call() file: mesh_tensorflow/transformer/evolved_transformer.py start line: 0 end line: 0 size: 17 LOC McCabe index: 1 number of parameters: 4 id: 289 unit: def update_context() file: mesh_tensorflow/transformer/funnel_transformer.py start line: 0 end line: 0 size: 17 LOC McCabe index: 1 number of parameters: 4 id: 290 unit: def _compute_softmax() file: mesh_tensorflow/bert/run_squad.py start line: 0 end line: 0 size: 17 LOC McCabe index: 7 number of parameters: 1 id: 291 unit: def get_train_examples() file: mesh_tensorflow/bert/run_classifier.py start line: 0 end line: 0 size: 17 LOC McCabe index: 4 number of parameters: 2 id: 292 unit: def convert_to_unicode() file: mesh_tensorflow/bert/tokenization.py start line: 0 end line: 0 size: 17 LOC McCabe index: 7 number of parameters: 1 id: 293 unit: def printable_text() file: mesh_tensorflow/bert/tokenization.py start line: 0 end line: 0 size: 17 LOC McCabe index: 9 number of parameters: 1 id: 294 unit: def _run_split_on_punc() file: mesh_tensorflow/bert/tokenization.py start line: 0 end line: 0 size: 17 LOC McCabe index: 5 number of parameters: 2 id: 295 unit: def conv2d_transpose() file: mesh_tensorflow/layers.py start line: 0 end line: 0 size: 16 LOC McCabe index: 3 number of parameters: 4 id: 296 unit: def conv3d() file: mesh_tensorflow/layers.py start line: 0 end line: 0 size: 16 LOC McCabe index: 3 number of parameters: 5 id: 297 unit: def local_1d_halo_exchange() file: mesh_tensorflow/layers.py start line: 0 end line: 0 size: 16 LOC McCabe index: 4 number of parameters: 5 id: 298 unit: def allconcat() file: mesh_tensorflow/simd_mesh_impl.py start line: 0 end line: 0 size: 16 LOC McCabe index: 4 number of parameters: 5 id: 299 unit: def alltoall() file: mesh_tensorflow/simd_mesh_impl.py start line: 0 end line: 0 size: 16 LOC McCabe index: 2 number of parameters: 5 id: 300 unit: def save_to_tfrecord() file: mesh_tensorflow/experimental/offline_data_aug.py start line: 0 end line: 0 size: 16 LOC McCabe index: 1 number of parameters: 6 id: 301 unit: def _collective_with_groups() file: mesh_tensorflow/placement_mesh_impl.py start line: 0 end line: 0 size: 16 LOC McCabe index: 7 number of parameters: 4 id: 302 unit: def slice_shape() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 16 LOC McCabe index: 4 number of parameters: 2 id: 303 unit: def minimum() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 16 LOC McCabe index: 1 number of parameters: 4 id: 304 unit: def maximum() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 16 LOC McCabe index: 1 number of parameters: 4 id: 305 unit: def sublayer_scale_norm() file: mesh_tensorflow/transformer/transformer.py start line: 0 end line: 0 size: 16 LOC McCabe index: 1 number of parameters: 5 id: 306 unit: def _match_ckpt_to_local_var_name() file: mesh_tensorflow/transformer/utils.py start line: 0 end line: 0 size: 16 LOC McCabe index: 12 number of parameters: 3 id: 307 unit: def __init__() file: mesh_tensorflow/transformer/transformer_layers.py start line: 0 end line: 0 size: 16 LOC McCabe index: 1 number of parameters: 8 id: 308 unit: def slanted_triangular() file: mesh_tensorflow/transformer/learning_rate_schedules.py start line: 0 end line: 0 size: 16 LOC McCabe index: 2 number of parameters: 6 id: 309 unit: def get_activation() file: mesh_tensorflow/bert/bert.py start line: 0 end line: 0 size: 16 LOC McCabe index: 1 number of parameters: 1 id: 310 unit: def _check_is_max_context() file: mesh_tensorflow/bert/run_squad.py start line: 0 end line: 0 size: 16 LOC McCabe index: 16 number of parameters: 3 id: 311 unit: def get_dev_examples() file: mesh_tensorflow/bert/run_classifier.py start line: 0 end line: 0 size: 16 LOC McCabe index: 4 number of parameters: 2 id: 312 unit: def tokenize() file: mesh_tensorflow/bert/tokenization.py start line: 0 end line: 0 size: 16 LOC McCabe index: 6 number of parameters: 2 id: 313 unit: def compute_topk_scores_and_seq() file: mesh_tensorflow/beam_search.py start line: 0 end line: 0 size: 15 LOC McCabe index: 3 number of parameters: 6 id: 314 unit: def get_tensor_num_entries() file: mesh_tensorflow/auto_mtf/graph_interface.py start line: 0 end line: 0 size: 15 LOC McCabe index: 5 number of parameters: 4 id: 315 unit: def get_slices() file: mesh_tensorflow/experimental/input_reader.py start line: 0 end line: 0 size: 15 LOC McCabe index: 5 number of parameters: 1 id: 316 unit: def _enqueue_laidout_tensors() file: mesh_tensorflow/experimental/input_reader.py start line: 0 end line: 0 size: 15 LOC McCabe index: 3 number of parameters: 2 id: 317 unit: def _rand_noise() file: mesh_tensorflow/experimental/data_aug_lib.py start line: 0 end line: 0 size: 15 LOC McCabe index: 5 number of parameters: 4 id: 318 unit: def receive() file: mesh_tensorflow/placement_mesh_impl.py start line: 0 end line: 0 size: 15 LOC McCabe index: 3 number of parameters: 4 id: 319 unit: def device_function() file: mesh_tensorflow/utils.py start line: 0 end line: 0 size: 15 LOC McCabe index: 2 number of parameters: 2 id: 320 unit: def cwise() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 15 LOC McCabe index: 2 number of parameters: 5 id: 321 unit: def _pool_helper() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 15 LOC McCabe index: 1 number of parameters: 3 id: 322 unit: def __init__() file: mesh_tensorflow/transformer/transformer.py start line: 0 end line: 0 size: 15 LOC McCabe index: 1 number of parameters: 8 id: 323 unit: def record_states_first_part_mode() file: mesh_tensorflow/transformer/transformer_layers.py start line: 0 end line: 0 size: 15 LOC McCabe index: 1 number of parameters: 5 id: 324 unit: def dense_product_fixup() file: mesh_tensorflow/transformer/fixup_layers.py start line: 0 end line: 0 size: 15 LOC McCabe index: 1 number of parameters: 7 id: 325 unit: def __init__() file: mesh_tensorflow/transformer/evolved_transformer.py start line: 0 end line: 0 size: 15 LOC McCabe index: 1 number of parameters: 5 id: 326 unit: def __init__() file: mesh_tensorflow/bert/run_squad.py start line: 0 end line: 0 size: 15 LOC McCabe index: 1 number of parameters: 8 id: 327 unit: def _create_examples() file: mesh_tensorflow/bert/run_classifier.py start line: 0 end line: 0 size: 15 LOC McCabe index: 4 number of parameters: 3 id: 328 unit: def _create_examples() file: mesh_tensorflow/bert/run_classifier.py start line: 0 end line: 0 size: 15 LOC McCabe index: 4 number of parameters: 3 id: 329 unit: def _create_examples() file: mesh_tensorflow/bert/run_classifier.py start line: 0 end line: 0 size: 15 LOC McCabe index: 5 number of parameters: 3 id: 330 unit: def conv1d() file: mesh_tensorflow/layers.py start line: 0 end line: 0 size: 14 LOC McCabe index: 1 number of parameters: 5 id: 331 unit: def conv2d() file: mesh_tensorflow/layers.py start line: 0 end line: 0 size: 14 LOC McCabe index: 3 number of parameters: 4 id: 332 unit: def apply_data_aug() file: mesh_tensorflow/experimental/offline_data_aug.py start line: 0 end line: 0 size: 14 LOC McCabe index: 3 number of parameters: 0 id: 333 unit: def finish() file: mesh_tensorflow/experimental/unet.py start line: 0 end line: 0 size: 14 LOC McCabe index: 2 number of parameters: 1 id: 334 unit: def intensity_shift() file: mesh_tensorflow/experimental/data_aug_lib.py start line: 0 end line: 0 size: 14 LOC McCabe index: 3 number of parameters: 4 id: 335 unit: def lower() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 14 LOC McCabe index: 8 number of parameters: 2 id: 336 unit: def __init__() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 14 LOC McCabe index: 3 number of parameters: 7 id: 337 unit: def __init__() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 14 LOC McCabe index: 3 number of parameters: 8 id: 338 unit: def _legacy_init() file: mesh_tensorflow/transformer/transformer.py start line: 0 end line: 0 size: 14 LOC McCabe index: 4 number of parameters: 3 id: 339 unit: def set_default_moe_hparams() file: mesh_tensorflow/transformer/moe.py start line: 0 end line: 0 size: 14 LOC McCabe index: 5 number of parameters: 1 id: 340 unit: def score_from_dataset() file: mesh_tensorflow/transformer/utils.py start line: 0 end line: 0 size: 14 LOC McCabe index: 1 number of parameters: 10 id: 341 unit: def learning_rate_schedule_noam() file: mesh_tensorflow/transformer/learning_rate_schedules.py start line: 0 end line: 0 size: 14 LOC McCabe index: 2 number of parameters: 6 id: 342 unit: def get_next_sentence_output() file: mesh_tensorflow/bert/bert.py start line: 0 end line: 0 size: 14 LOC McCabe index: 2 number of parameters: 2 id: 343 unit: def __sub__() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 13 LOC McCabe index: 7 number of parameters: 2 id: 344 unit: def allsplit() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 13 LOC McCabe index: 6 number of parameters: 5 id: 345 unit: def __init__() file: mesh_tensorflow/optimize.py start line: 0 end line: 0 size: 13 LOC McCabe index: 1 number of parameters: 7 id: 346 unit: def encode_all_features() file: mesh_tensorflow/transformer/dataset.py start line: 0 end line: 0 size: 13 LOC McCabe index: 3 number of parameters: 2 id: 347 unit: def trim_and_pad_dataset() file: mesh_tensorflow/transformer/dataset.py start line: 0 end line: 0 size: 13 LOC McCabe index: 5 number of parameters: 3 id: 348 unit: def get_layer_timing_signal_learned_1d() file: mesh_tensorflow/transformer/universal_transformer.py start line: 0 end line: 0 size: 13 LOC McCabe index: 1 number of parameters: 5 id: 349 unit: def _split_into_groups() file: mesh_tensorflow/transformer/moe.py start line: 0 end line: 0 size: 13 LOC McCabe index: 4 number of parameters: 3 id: 350 unit: def __init__() file: mesh_tensorflow/transformer/transformer_layers.py start line: 0 end line: 0 size: 13 LOC McCabe index: 1 number of parameters: 7 id: 351 unit: def attention_internal() file: mesh_tensorflow/transformer/transformer_layers.py start line: 0 end line: 0 size: 13 LOC McCabe index: 2 number of parameters: 6 id: 352 unit: def record_states_incremental_mode() file: mesh_tensorflow/transformer/transformer_layers.py start line: 0 end line: 0 size: 13 LOC McCabe index: 2 number of parameters: 5 id: 353 unit: def __init__() file: mesh_tensorflow/transformer/transformer_layers.py start line: 0 end line: 0 size: 13 LOC McCabe index: 1 number of parameters: 7 id: 354 unit: def from_dict() file: mesh_tensorflow/bert/bert.py start line: 0 end line: 0 size: 13 LOC McCabe index: 4 number of parameters: 2 id: 355 unit: def feedforward() file: mesh_tensorflow/bert/bert.py start line: 0 end line: 0 size: 13 LOC McCabe index: 1 number of parameters: 2 id: 356 unit: def __repr__() file: mesh_tensorflow/bert/run_squad.py start line: 0 end line: 0 size: 13 LOC McCabe index: 4 number of parameters: 1 id: 357 unit: def multihead_encdec_attention_incremental() file: mesh_tensorflow/layers.py start line: 0 end line: 0 size: 12 LOC McCabe index: 1 number of parameters: 7 id: 358 unit: def __init__() file: mesh_tensorflow/experimental/input_reader.py start line: 0 end line: 0 size: 12 LOC McCabe index: 1 number of parameters: 8 id: 359 unit: def start_infeed_thread() file: mesh_tensorflow/experimental/input_reader.py start line: 0 end line: 0 size: 12 LOC McCabe index: 4 number of parameters: 4 id: 360 unit: def image_corruption() file: mesh_tensorflow/experimental/data_aug_lib.py start line: 0 end line: 0 size: 12 LOC McCabe index: 3 number of parameters: 5 id: 361 unit: def convert_to_dimension() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 12 LOC McCabe index: 7 number of parameters: 1 id: 362 unit: def slice_begin() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 12 LOC McCabe index: 3 number of parameters: 3 id: 363 unit: def shift_by_n_processors() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 12 LOC McCabe index: 4 number of parameters: 5 id: 364 unit: def apply_grad() file: mesh_tensorflow/optimize.py start line: 0 end line: 0 size: 12 LOC McCabe index: 2 number of parameters: 3 id: 365 unit: def sublayer_rms_norm() file: mesh_tensorflow/transformer/transformer.py start line: 0 end line: 0 size: 12 LOC McCabe index: 1 number of parameters: 5 id: 366 unit: def shift_targets() file: mesh_tensorflow/transformer/transformer.py start line: 0 end line: 0 size: 12 LOC McCabe index: 2 number of parameters: 3 id: 367 unit: def compute_loss() file: mesh_tensorflow/transformer/adaptive_softmax.py start line: 0 end line: 0 size: 12 LOC McCabe index: 1 number of parameters: 5 id: 368 unit: def _hidden_to_logits() file: mesh_tensorflow/transformer/adaptive_softmax.py start line: 0 end line: 0 size: 12 LOC McCabe index: 2 number of parameters: 3 id: 369 unit: def score_with_estimator() file: mesh_tensorflow/transformer/utils.py start line: 0 end line: 0 size: 12 LOC McCabe index: 6 number of parameters: 7 id: 370 unit: def compute_q() file: mesh_tensorflow/transformer/transformer_layers.py start line: 0 end line: 0 size: 12 LOC McCabe index: 2 number of parameters: 3 id: 371 unit: def call() file: mesh_tensorflow/transformer/transformer_layers.py start line: 0 end line: 0 size: 12 LOC McCabe index: 4 number of parameters: 4 id: 372 unit: def attention_params_simple() file: mesh_tensorflow/transformer/attention.py start line: 0 end line: 0 size: 12 LOC McCabe index: 1 number of parameters: 5 id: 373 unit: def _dense_var_to_tensor() file: mesh_tensorflow/tpu_variables.py start line: 0 end line: 0 size: 12 LOC McCabe index: 6 number of parameters: 4 id: 374 unit: def summary_for_clip_activation_gradient() file: mesh_tensorflow/layers.py start line: 0 end line: 0 size: 11 LOC McCabe index: 2 number of parameters: 3 id: 375 unit: def attention_bias_local_block() file: mesh_tensorflow/layers.py start line: 0 end line: 0 size: 11 LOC McCabe index: 1 number of parameters: 4 id: 376 unit: def compress_mean() file: mesh_tensorflow/layers.py start line: 0 end line: 0 size: 11 LOC McCabe index: 1 number of parameters: 3 id: 377 unit: def _create_group_assignment() file: mesh_tensorflow/simd_mesh_impl.py start line: 0 end line: 0 size: 11 LOC McCabe index: 4 number of parameters: 2 id: 378 unit: def __init__() file: mesh_tensorflow/auto_mtf/layout_optimizer.py start line: 0 end line: 0 size: 11 LOC McCabe index: 1 number of parameters: 3 id: 379 unit: def _initialize_splittable_dimensions() file: mesh_tensorflow/auto_mtf/valid_layouts.py start line: 0 end line: 0 size: 11 LOC McCabe index: 6 number of parameters: 2 id: 380 unit: def autostack() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 11 LOC McCabe index: 3 number of parameters: 1 id: 381 unit: def laid_out_slice_num() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 11 LOC McCabe index: 3 number of parameters: 2 id: 382 unit: def make_slices() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 11 LOC McCabe index: 2 number of parameters: 3 id: 383 unit: def gelu() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 11 LOC McCabe index: 1 number of parameters: 1 id: 384 unit: def concat() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 11 LOC McCabe index: 3 number of parameters: 3 id: 385 unit: def pack_or_pad() file: mesh_tensorflow/transformer/dataset.py start line: 0 end line: 0 size: 11 LOC McCabe index: 5 number of parameters: 5 id: 386 unit: def step_preprocess() file: mesh_tensorflow/transformer/universal_transformer.py start line: 0 end line: 0 size: 11 LOC McCabe index: 5 number of parameters: 4 id: 387 unit: def get_inputs_from_file() file: mesh_tensorflow/transformer/utils.py start line: 0 end line: 0 size: 11 LOC McCabe index: 7 number of parameters: 2 id: 388 unit: def compute_q() file: mesh_tensorflow/transformer/transformer_layers.py start line: 0 end line: 0 size: 11 LOC McCabe index: 2 number of parameters: 3 id: 389 unit: def compute_output() file: mesh_tensorflow/transformer/attention.py start line: 0 end line: 0 size: 11 LOC McCabe index: 3 number of parameters: 3 id: 390 unit: def __init__() file: mesh_tensorflow/bert/run_classifier.py start line: 0 end line: 0 size: 11 LOC McCabe index: 1 number of parameters: 6 id: 391 unit: def load_vocab() file: mesh_tensorflow/bert/tokenization.py start line: 0 end line: 0 size: 11 LOC McCabe index: 4 number of parameters: 1 id: 392 unit: def _tokenize_chinese_chars() file: mesh_tensorflow/bert/tokenization.py start line: 0 end line: 0 size: 11 LOC McCabe index: 3 number of parameters: 2 id: 393 unit: def _is_chinese_char() file: mesh_tensorflow/bert/tokenization.py start line: 0 end line: 0 size: 11 LOC McCabe index: 20 number of parameters: 2 id: 394 unit: def _clean_text() file: mesh_tensorflow/bert/tokenization.py start line: 0 end line: 0 size: 11 LOC McCabe index: 6 number of parameters: 2 id: 395 unit: def corr() file: mesh_tensorflow/layers.py start line: 0 end line: 0 size: 10 LOC McCabe index: 1 number of parameters: 4 id: 396 unit: def assign_to_slices() file: mesh_tensorflow/simd_mesh_impl.py start line: 0 end line: 0 size: 10 LOC McCabe index: 3 number of parameters: 4 id: 397 unit: def _generate_assignments() file: mesh_tensorflow/auto_mtf/layout_optimizer.py start line: 0 end line: 0 size: 10 LOC McCabe index: 4 number of parameters: 2 id: 398 unit: def _initialize_mtf_dimension_name_to_size_gcd() file: mesh_tensorflow/auto_mtf/valid_layouts.py start line: 0 end line: 0 size: 10 LOC McCabe index: 4 number of parameters: 2 id: 399 unit: def _slice_tensor() file: mesh_tensorflow/experimental/input_reader.py start line: 0 end line: 0 size: 10 LOC McCabe index: 3 number of parameters: 4 id: 400 unit: def slice_is_master() file: mesh_tensorflow/placement_mesh_impl.py start line: 0 end line: 0 size: 10 LOC McCabe index: 5 number of parameters: 1 id: 401 unit: def slicewise() file: mesh_tensorflow/placement_mesh_impl.py start line: 0 end line: 0 size: 10 LOC McCabe index: 5 number of parameters: 3 id: 402 unit: def tensor_layout() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 10 LOC McCabe index: 5 number of parameters: 3 id: 403 unit: def slicewise_delay_allreduce() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 10 LOC McCabe index: 3 number of parameters: 3 id: 404 unit: def __init__() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 10 LOC McCabe index: 1 number of parameters: 5 id: 405 unit: def elu() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 10 LOC McCabe index: 1 number of parameters: 1 id: 406 unit: def selu() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 10 LOC McCabe index: 1 number of parameters: 1 id: 407 unit: def __init__() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 10 LOC McCabe index: 3 number of parameters: 4 id: 408 unit: def lower() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 10 LOC McCabe index: 3 number of parameters: 2 id: 409 unit: def _factored_dims() file: mesh_tensorflow/optimize.py start line: 0 end line: 0 size: 10 LOC McCabe index: 6 number of parameters: 2 id: 410 unit: def __init__() file: mesh_tensorflow/optimize.py start line: 0 end line: 0 size: 10 LOC McCabe index: 1 number of parameters: 4 id: 411 unit: def __init__() file: mesh_tensorflow/optimize.py start line: 0 end line: 0 size: 10 LOC McCabe index: 1 number of parameters: 4 id: 412 unit: def sample_from_text_line_datasets() file: mesh_tensorflow/transformer/dataset.py start line: 0 end line: 0 size: 10 LOC McCabe index: 2 number of parameters: 3 id: 413 unit: def autoregressive_inputs() file: mesh_tensorflow/transformer/transformer.py start line: 0 end line: 0 size: 10 LOC McCabe index: 2 number of parameters: 2 id: 414 unit: def parse_gin_defaults_and_flags() file: mesh_tensorflow/transformer/utils.py start line: 0 end line: 0 size: 10 LOC McCabe index: 3 number of parameters: 2 id: 415 unit: def metric_max() file: mesh_tensorflow/transformer/utils.py start line: 0 end line: 0 size: 10 LOC McCabe index: 1 number of parameters: 3 id: 416 unit: def make_params() file: mesh_tensorflow/transformer/transformer_layers.py start line: 0 end line: 0 size: 10 LOC McCabe index: 1 number of parameters: 2 id: 417 unit: def __init__() file: mesh_tensorflow/transformer/transformer_layers.py start line: 0 end line: 0 size: 10 LOC McCabe index: 2 number of parameters: 3 id: 418 unit: def __init__() file: mesh_tensorflow/transformer/memory_layers.py start line: 0 end line: 0 size: 10 LOC McCabe index: 1 number of parameters: 5 id: 419 unit: def sublayer_fixup_shift() file: mesh_tensorflow/transformer/fixup_layers.py start line: 0 end line: 0 size: 10 LOC McCabe index: 1 number of parameters: 3 id: 420 unit: def assign_add() file: mesh_tensorflow/tpu_variables.py start line: 0 end line: 0 size: 10 LOC McCabe index: 2 number of parameters: 5 id: 421 unit: def assign_sub() file: mesh_tensorflow/tpu_variables.py start line: 0 end line: 0 size: 10 LOC McCabe index: 2 number of parameters: 5 id: 422 unit: def size_per_head_dim() file: mesh_tensorflow/bert/bert.py start line: 0 end line: 0 size: 10 LOC McCabe index: 3 number of parameters: 1 id: 423 unit: def convert_examples_to_features() file: mesh_tensorflow/bert/run_classifier.py start line: 0 end line: 0 size: 10 LOC McCabe index: 3 number of parameters: 4 id: 424 unit: def get_operation_mtf_dimension_names() file: mesh_tensorflow/auto_mtf/graph_interface.py start line: 0 end line: 0 size: 9 LOC McCabe index: 3 number of parameters: 2 id: 425 unit: def _compute_initial_out_degree() file: mesh_tensorflow/auto_mtf/graph_interface.py start line: 0 end line: 0 size: 9 LOC McCabe index: 5 number of parameters: 1 id: 426 unit: def minimize_peak_memory() file: mesh_tensorflow/auto_mtf/scheduler.py start line: 0 end line: 0 size: 9 LOC McCabe index: 3 number of parameters: 2 id: 427 unit: def _local_var_name() file: mesh_tensorflow/auto_mtf/layout_optimizer.py start line: 0 end line: 0 size: 9 LOC McCabe index: 3 number of parameters: 2 id: 428 unit: def _get_scaffold() file: mesh_tensorflow/experimental/model_executor.py start line: 0 end line: 0 size: 9 LOC McCabe index: 1 number of parameters: 1 id: 429 unit: def Print() file: mesh_tensorflow/placement_mesh_impl.py start line: 0 end line: 0 size: 9 LOC McCabe index: 3 number of parameters: 5 id: 430 unit: def tensor_dimension_to_mesh_axis() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 9 LOC McCabe index: 5 number of parameters: 3 id: 431 unit: def copy_masters_to_slices() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 9 LOC McCabe index: 4 number of parameters: 1 id: 432 unit: def binary_op_with_broadcasting() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 9 LOC McCabe index: 2 number of parameters: 5 id: 433 unit: def gradient() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 9 LOC McCabe index: 4 number of parameters: 2 id: 434 unit: def ensure_dataset_eos() file: mesh_tensorflow/transformer/dataset.py start line: 0 end line: 0 size: 9 LOC McCabe index: 4 number of parameters: 2 id: 435 unit: def _layer_fn() file: mesh_tensorflow/transformer/transformer.py start line: 0 end line: 0 size: 9 LOC McCabe index: 2 number of parameters: 4 id: 436 unit: def loss_denominator() file: mesh_tensorflow/transformer/transformer.py start line: 0 end line: 0 size: 9 LOC McCabe index: 3 number of parameters: 3 id: 437 unit: def ids_to_embedding() file: mesh_tensorflow/transformer/adaptive_softmax.py start line: 0 end line: 0 size: 9 LOC McCabe index: 2 number of parameters: 3 id: 438 unit: def adaptive_softmax_loss_fn() file: mesh_tensorflow/transformer/adaptive_softmax.py start line: 0 end line: 0 size: 9 LOC McCabe index: 1 number of parameters: 5 id: 439 unit: def ut_basic() file: mesh_tensorflow/transformer/universal_transformer.py start line: 0 end line: 0 size: 9 LOC McCabe index: 3 number of parameters: 4 id: 440 unit: def _router_z_loss() file: mesh_tensorflow/transformer/moe.py start line: 0 end line: 0 size: 9 LOC McCabe index: 2 number of parameters: 4 id: 441 unit: def metric_sum() file: mesh_tensorflow/transformer/utils.py start line: 0 end line: 0 size: 9 LOC McCabe index: 1 number of parameters: 3 id: 442 unit: def make_params() file: mesh_tensorflow/transformer/transformer_layers.py start line: 0 end line: 0 size: 9 LOC McCabe index: 1 number of parameters: 2 id: 443 unit: def _sigmoid_tree() file: mesh_tensorflow/transformer/vocab_embeddings.py start line: 0 end line: 0 size: 9 LOC McCabe index: 1 number of parameters: 2 id: 444 unit: def __init__() file: mesh_tensorflow/transformer/fixup_layers.py start line: 0 end line: 0 size: 9 LOC McCabe index: 1 number of parameters: 5 id: 445 unit: def call() file: mesh_tensorflow/transformer/evolved_transformer.py start line: 0 end line: 0 size: 9 LOC McCabe index: 1 number of parameters: 4 id: 446 unit: def _pad_channels_dim() file: mesh_tensorflow/transformer/evolved_transformer.py start line: 0 end line: 0 size: 9 LOC McCabe index: 3 number of parameters: 2 id: 447 unit: def _layer_fn() file: mesh_tensorflow/transformer/funnel_transformer.py start line: 0 end line: 0 size: 9 LOC McCabe index: 2 number of parameters: 4 id: 448 unit: def assign() file: mesh_tensorflow/tpu_variables.py start line: 0 end line: 0 size: 9 LOC McCabe index: 2 number of parameters: 5 id: 449 unit: def _improve_answer_span() file: mesh_tensorflow/bert/run_squad.py start line: 0 end line: 0 size: 9 LOC McCabe index: 4 number of parameters: 5 id: 450 unit: def _truncate_seq_pair() file: mesh_tensorflow/bert/run_classifier.py start line: 0 end line: 0 size: 9 LOC McCabe index: 5 number of parameters: 3 id: 451 unit: def tokenize() file: mesh_tensorflow/bert/tokenization.py start line: 0 end line: 0 size: 9 LOC McCabe index: 4 number of parameters: 2 id: 452 unit: def _run_strip_accents() file: mesh_tensorflow/bert/tokenization.py start line: 0 end line: 0 size: 9 LOC McCabe index: 3 number of parameters: 2 id: 453 unit: def _is_punctuation() file: mesh_tensorflow/bert/tokenization.py start line: 0 end line: 0 size: 9 LOC McCabe index: 12 number of parameters: 1 id: 454 unit: def sigmoid_cross_entropy_with_logits() file: mesh_tensorflow/layers.py start line: 0 end line: 0 size: 8 LOC McCabe index: 2 number of parameters: 2 id: 455 unit: def pnum_tensor() file: mesh_tensorflow/simd_mesh_impl.py start line: 0 end line: 0 size: 8 LOC McCabe index: 2 number of parameters: 1 id: 456 unit: def _initialize_operations() file: mesh_tensorflow/auto_mtf/graph_interface.py start line: 0 end line: 0 size: 8 LOC McCabe index: 3 number of parameters: 1 id: 457 unit: def gpu_placement() file: mesh_tensorflow/experimental/input_reader.py start line: 0 end line: 0 size: 8 LOC McCabe index: 1 number of parameters: 2 id: 458 unit: def _print_variable_values() file: mesh_tensorflow/experimental/model_executor.py start line: 0 end line: 0 size: 8 LOC McCabe index: 2 number of parameters: 1 id: 459 unit: def convert_to_shape() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 8 LOC McCabe index: 4 number of parameters: 1 id: 460 unit: def verify_slice_shapes() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 8 LOC McCabe index: 2 number of parameters: 3 id: 461 unit: def gradient() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 8 LOC McCabe index: 1 number of parameters: 2 id: 462 unit: def _do_use_weight_decay() file: mesh_tensorflow/optimize.py start line: 0 end line: 0 size: 8 LOC McCabe index: 5 number of parameters: 2 id: 463 unit: def product_learning_rate() file: mesh_tensorflow/optimize.py start line: 0 end line: 0 size: 8 LOC McCabe index: 3 number of parameters: 4 id: 464 unit: def nonpadding() file: mesh_tensorflow/transformer/transformer.py start line: 0 end line: 0 size: 8 LOC McCabe index: 3 number of parameters: 1 id: 465 unit: def sublayer_dropout() file: mesh_tensorflow/transformer/transformer.py start line: 0 end line: 0 size: 8 LOC McCabe index: 3 number of parameters: 4 id: 466 unit: def decode() file: mesh_tensorflow/transformer/transformer.py start line: 0 end line: 0 size: 8 LOC McCabe index: 3 number of parameters: 3 id: 467 unit: def get_log_softmax_prefix() file: mesh_tensorflow/transformer/adaptive_softmax.py start line: 0 end line: 0 size: 8 LOC McCabe index: 1 number of parameters: 3 id: 468 unit: def _dropout() file: mesh_tensorflow/transformer/universal_transformer.py start line: 0 end line: 0 size: 8 LOC McCabe index: 3 number of parameters: 3 id: 469 unit: def get_variable_dtype() file: mesh_tensorflow/transformer/utils.py start line: 0 end line: 0 size: 8 LOC McCabe index: 1 number of parameters: 3 id: 470 unit: def __init__() file: mesh_tensorflow/transformer/transformer_layers.py start line: 0 end line: 0 size: 8 LOC McCabe index: 2 number of parameters: 2 id: 471 unit: def hidden_to_logits() file: mesh_tensorflow/transformer/vocab_embeddings.py start line: 0 end line: 0 size: 8 LOC McCabe index: 3 number of parameters: 3 id: 472 unit: def _dropout() file: mesh_tensorflow/transformer/vocab_embeddings.py start line: 0 end line: 0 size: 8 LOC McCabe index: 3 number of parameters: 3 id: 473 unit: def get_single_scalar_bias() file: mesh_tensorflow/transformer/fixup_layers.py start line: 0 end line: 0 size: 8 LOC McCabe index: 1 number of parameters: 2 id: 474 unit: def sublayer_fixup_scale() file: mesh_tensorflow/transformer/fixup_layers.py start line: 0 end line: 0 size: 8 LOC McCabe index: 1 number of parameters: 3 id: 475 unit: def product_learning_rate() file: mesh_tensorflow/transformer/learning_rate_schedules.py start line: 0 end line: 0 size: 8 LOC McCabe index: 3 number of parameters: 4 id: 476 unit: def linear_decay() file: mesh_tensorflow/transformer/learning_rate_schedules.py start line: 0 end line: 0 size: 8 LOC McCabe index: 2 number of parameters: 3 id: 477 unit: def linear_warmup() file: mesh_tensorflow/transformer/learning_rate_schedules.py start line: 0 end line: 0 size: 8 LOC McCabe index: 2 number of parameters: 3 id: 478 unit: def linear_decay_learning_rate() file: mesh_tensorflow/transformer/learning_rate_schedules.py start line: 0 end line: 0 size: 8 LOC McCabe index: 1 number of parameters: 4 id: 479 unit: def _dropout() file: mesh_tensorflow/transformer/evolved_transformer.py start line: 0 end line: 0 size: 8 LOC McCabe index: 3 number of parameters: 3 id: 480 unit: def compute_q() file: mesh_tensorflow/transformer/attention.py start line: 0 end line: 0 size: 8 LOC McCabe index: 3 number of parameters: 2 id: 481 unit: def compute_kv() file: mesh_tensorflow/transformer/attention.py start line: 0 end line: 0 size: 8 LOC McCabe index: 3 number of parameters: 2 id: 482 unit: def compute_k() file: mesh_tensorflow/transformer/attention.py start line: 0 end line: 0 size: 8 LOC McCabe index: 3 number of parameters: 2 id: 483 unit: def compute_v() file: mesh_tensorflow/transformer/attention.py start line: 0 end line: 0 size: 8 LOC McCabe index: 3 number of parameters: 2 id: 484 unit: def _decode_record() file: mesh_tensorflow/bert/run_pretraining.py start line: 0 end line: 0 size: 8 LOC McCabe index: 3 number of parameters: 2 id: 485 unit: def _get_best_indexes() file: mesh_tensorflow/bert/run_squad.py start line: 0 end line: 0 size: 8 LOC McCabe index: 3 number of parameters: 2 id: 486 unit: def us_einsum() file: mesh_tensorflow/layers.py start line: 0 end line: 0 size: 7 LOC McCabe index: 5 number of parameters: 3 id: 487 unit: def multihead_attention_vars() file: mesh_tensorflow/layers.py start line: 0 end line: 0 size: 7 LOC McCabe index: 1 number of parameters: 7 id: 488 unit: def __init__() file: mesh_tensorflow/auto_mtf/graph_interface.py start line: 0 end line: 0 size: 7 LOC McCabe index: 2 number of parameters: 3 id: 489 unit: def _get_memory_contents() file: mesh_tensorflow/auto_mtf/layout_optimizer.py start line: 0 end line: 0 size: 7 LOC McCabe index: 2 number of parameters: 1 id: 490 unit: def __init__() file: mesh_tensorflow/auto_mtf/valid_layouts.py start line: 0 end line: 0 size: 7 LOC McCabe index: 1 number of parameters: 3 id: 491 unit: def after_create_session() file: mesh_tensorflow/experimental/model_executor.py start line: 0 end line: 0 size: 7 LOC McCabe index: 3 number of parameters: 3 id: 492 unit: def random() file: mesh_tensorflow/placement_mesh_impl.py start line: 0 end line: 0 size: 7 LOC McCabe index: 1 number of parameters: 4 id: 493 unit: def alltoall_pointtwise() file: mesh_tensorflow/placement_mesh_impl.py start line: 0 end line: 0 size: 7 LOC McCabe index: 2 number of parameters: 4 id: 494 unit: def topology_rank() file: mesh_tensorflow/utils.py start line: 0 end line: 0 size: 7 LOC McCabe index: 4 number of parameters: 1 id: 495 unit: def rename_dimension() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 7 LOC McCabe index: 4 number of parameters: 3 id: 496 unit: def resize_dimension() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 7 LOC McCabe index: 4 number of parameters: 3 id: 497 unit: def __init__() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 7 LOC McCabe index: 2 number of parameters: 1 id: 498 unit: def make_variables_untrainable() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 7 LOC McCabe index: 4 number of parameters: 2 id: 499 unit: def to_laid_out_tensor() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 7 LOC McCabe index: 3 number of parameters: 1 id: 500 unit: def leaky_relu() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 7 LOC McCabe index: 1 number of parameters: 3 id: 501 unit: def lower() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 7 LOC McCabe index: 1 number of parameters: 2 id: 502 unit: def make_optimizer() file: mesh_tensorflow/optimize.py start line: 0 end line: 0 size: 7 LOC McCabe index: 3 number of parameters: 2 id: 503 unit: def apply_grads() file: mesh_tensorflow/optimize.py start line: 0 end line: 0 size: 7 LOC McCabe index: 3 number of parameters: 3 id: 504 unit: def _learning_rate_default() file: mesh_tensorflow/optimize.py start line: 0 end line: 0 size: 7 LOC McCabe index: 3 number of parameters: 2 id: 505 unit: def __init__() file: mesh_tensorflow/transformer/t2t_vocabulary.py start line: 0 end line: 0 size: 7 LOC McCabe index: 2 number of parameters: 2 id: 506 unit: def supervised_to_dict() file: mesh_tensorflow/transformer/dataset.py start line: 0 end line: 0 size: 7 LOC McCabe index: 2 number of parameters: 2 id: 507 unit: def sublayer_mask_padding() file: mesh_tensorflow/transformer/transformer.py start line: 0 end line: 0 size: 7 LOC McCabe index: 2 number of parameters: 3 id: 508 unit: def sublayer_rezero() file: mesh_tensorflow/transformer/transformer.py start line: 0 end line: 0 size: 7 LOC McCabe index: 1 number of parameters: 4 id: 509 unit: def __init__() file: mesh_tensorflow/transformer/adaptive_softmax.py start line: 0 end line: 0 size: 7 LOC McCabe index: 1 number of parameters: 6 id: 510 unit: def get_project_to_cluster_length() file: mesh_tensorflow/transformer/adaptive_softmax.py start line: 0 end line: 0 size: 7 LOC McCabe index: 1 number of parameters: 3 id: 511 unit: def write_lines_to_file() file: mesh_tensorflow/transformer/utils.py start line: 0 end line: 0 size: 7 LOC McCabe index: 3 number of parameters: 2 id: 512 unit: def compute_k() file: mesh_tensorflow/transformer/transformer_layers.py start line: 0 end line: 0 size: 7 LOC McCabe index: 1 number of parameters: 3 id: 513 unit: def compute_v() file: mesh_tensorflow/transformer/transformer_layers.py start line: 0 end line: 0 size: 7 LOC McCabe index: 1 number of parameters: 3 id: 514 unit: def compute_y() file: mesh_tensorflow/transformer/transformer_layers.py start line: 0 end line: 0 size: 7 LOC McCabe index: 1 number of parameters: 3 id: 515 unit: def compute_y() file: mesh_tensorflow/transformer/transformer_layers.py start line: 0 end line: 0 size: 7 LOC McCabe index: 1 number of parameters: 3 id: 516 unit: def _rearrange_sentinels() file: mesh_tensorflow/transformer/vocab_embeddings.py start line: 0 end line: 0 size: 7 LOC McCabe index: 2 number of parameters: 2 id: 517 unit: def clip_by_global_norm() file: mesh_tensorflow/bert/optimization.py start line: 0 end line: 0 size: 7 LOC McCabe index: 5 number of parameters: 2 id: 518 unit: def embedding_initializer() file: mesh_tensorflow/bert/bert.py start line: 0 end line: 0 size: 7 LOC McCabe index: 2 number of parameters: 1 id: 519 unit: def _read_tsv() file: mesh_tensorflow/bert/run_classifier.py start line: 0 end line: 0 size: 7 LOC McCabe index: 2 number of parameters: 3 id: 520 unit: def _is_whitespace() file: mesh_tensorflow/bert/tokenization.py start line: 0 end line: 0 size: 7 LOC McCabe index: 7 number of parameters: 1 id: 521 unit: def _is_control() file: mesh_tensorflow/bert/tokenization.py start line: 0 end line: 0 size: 7 LOC McCabe index: 5 number of parameters: 1 id: 522 unit: def __init__() file: mesh_tensorflow/layers.py start line: 0 end line: 0 size: 6 LOC McCabe index: 1 number of parameters: 4 id: 523 unit: def layout() file: mesh_tensorflow/auto_mtf/api.py start line: 0 end line: 0 size: 6 LOC McCabe index: 1 number of parameters: 3 id: 524 unit: def get_tensor_shape() file: mesh_tensorflow/auto_mtf/graph_interface.py start line: 0 end line: 0 size: 6 LOC McCabe index: 2 number of parameters: 2 id: 525 unit: def get_tensor_device() file: mesh_tensorflow/auto_mtf/graph_interface.py start line: 0 end line: 0 size: 6 LOC McCabe index: 2 number of parameters: 2 id: 526 unit: def get_operation_device() file: mesh_tensorflow/auto_mtf/graph_interface.py start line: 0 end line: 0 size: 6 LOC McCabe index: 2 number of parameters: 2 id: 527 unit: def get_tensor_mtf_dimension_names() file: mesh_tensorflow/auto_mtf/graph_interface.py start line: 0 end line: 0 size: 6 LOC McCabe index: 2 number of parameters: 2 id: 528 unit: def _initialize_tensor_name_to_ids() file: mesh_tensorflow/auto_mtf/graph_interface.py start line: 0 end line: 0 size: 6 LOC McCabe index: 3 number of parameters: 1 id: 529 unit: def __init__() file: mesh_tensorflow/auto_mtf/memory_estimator.py start line: 0 end line: 0 size: 6 LOC McCabe index: 1 number of parameters: 4 id: 530 unit: def _host_device_to_id() file: mesh_tensorflow/experimental/input_reader.py start line: 0 end line: 0 size: 6 LOC McCabe index: 1 number of parameters: 1 id: 531 unit: def _host_id_to_tf_device() file: mesh_tensorflow/experimental/input_reader.py start line: 0 end line: 0 size: 6 LOC McCabe index: 2 number of parameters: 2 id: 532 unit: def _validate_args() file: mesh_tensorflow/experimental/input_reader.py start line: 0 end line: 0 size: 6 LOC McCabe index: 1 number of parameters: 1 id: 533 unit: def capture() file: mesh_tensorflow/experimental/model_executor.py start line: 0 end line: 0 size: 6 LOC McCabe index: 2 number of parameters: 2 id: 534 unit: def get() file: mesh_tensorflow/experimental/model_executor.py start line: 0 end line: 0 size: 6 LOC McCabe index: 2 number of parameters: 1 id: 535 unit: def __init__() file: mesh_tensorflow/utils.py start line: 0 end line: 0 size: 6 LOC McCabe index: 2 number of parameters: 3 id: 536 unit: def remove_summaries() file: mesh_tensorflow/utils.py start line: 0 end line: 0 size: 6 LOC McCabe index: 1 number of parameters: 0 id: 537 unit: def __add__() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 6 LOC McCabe index: 3 number of parameters: 2 id: 538 unit: def get_dim_by_name() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 6 LOC McCabe index: 3 number of parameters: 2 id: 539 unit: def convert_to_layout_rules() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 6 LOC McCabe index: 3 number of parameters: 1 id: 540 unit: def laid_out_pcoord() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 6 LOC McCabe index: 3 number of parameters: 2 id: 541 unit: def broadcast_impl() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 6 LOC McCabe index: 1 number of parameters: 4 id: 542 unit: def compatible_lazy_allreduce_sums() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 6 LOC McCabe index: 1 number of parameters: 1 id: 543 unit: def cast() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 6 LOC McCabe index: 2 number of parameters: 3 id: 544 unit: def __init__() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 6 LOC McCabe index: 2 number of parameters: 4 id: 545 unit: def ids_to_embedding() file: mesh_tensorflow/transformer/transformer.py start line: 0 end line: 0 size: 6 LOC McCabe index: 2 number of parameters: 3 id: 546 unit: def hidden_to_logits() file: mesh_tensorflow/transformer/transformer.py start line: 0 end line: 0 size: 6 LOC McCabe index: 2 number of parameters: 3 id: 547 unit: def hidden_to_logits() file: mesh_tensorflow/transformer/adaptive_softmax.py start line: 0 end line: 0 size: 6 LOC McCabe index: 2 number of parameters: 3 id: 548 unit: def compute_log_likelihoods() file: mesh_tensorflow/transformer/utils.py start line: 0 end line: 0 size: 6 LOC McCabe index: 2 number of parameters: 3 id: 549 unit: def __init__() file: mesh_tensorflow/transformer/transformer_layers.py start line: 0 end line: 0 size: 6 LOC McCabe index: 1 number of parameters: 5 id: 550 unit: def attention_kwargs_from_context() file: mesh_tensorflow/transformer/transformer_layers.py start line: 0 end line: 0 size: 6 LOC McCabe index: 3 number of parameters: 2 id: 551 unit: def update_state() file: mesh_tensorflow/transformer/transformer_layers.py start line: 0 end line: 0 size: 6 LOC McCabe index: 1 number of parameters: 6 id: 552 unit: def truncated_rsqrt() file: mesh_tensorflow/transformer/learning_rate_schedules.py start line: 0 end line: 0 size: 6 LOC McCabe index: 1 number of parameters: 3 id: 553 unit: def call() file: mesh_tensorflow/transformer/evolved_transformer.py start line: 0 end line: 0 size: 6 LOC McCabe index: 1 number of parameters: 4 id: 554 unit: def _enclosing_tpu_context() file: mesh_tensorflow/tpu_variables.py start line: 0 end line: 0 size: 6 LOC McCabe index: 3 number of parameters: 0 id: 555 unit: def __init__() file: mesh_tensorflow/tpu_variables.py start line: 0 end line: 0 size: 6 LOC McCabe index: 1 number of parameters: 3 id: 556 unit: def _assign_dependencies() file: mesh_tensorflow/tpu_variables.py start line: 0 end line: 0 size: 6 LOC McCabe index: 2 number of parameters: 1 id: 557 unit: def dense_initializer() file: mesh_tensorflow/bert/bert.py start line: 0 end line: 0 size: 6 LOC McCabe index: 2 number of parameters: 1 id: 558 unit: def preserve_token() file: mesh_tensorflow/bert/tokenization.py start line: 0 end line: 0 size: 6 LOC McCabe index: 3 number of parameters: 2 id: 559 unit: def whitespace_tokenize() file: mesh_tensorflow/bert/tokenization.py start line: 0 end line: 0 size: 6 LOC McCabe index: 2 number of parameters: 1 id: 560 unit: def __init__() file: mesh_tensorflow/bert/tokenization.py start line: 0 end line: 0 size: 6 LOC McCabe index: 2 number of parameters: 3 id: 561 unit: def attention_mask_same_segment() file: mesh_tensorflow/layers.py start line: 0 end line: 0 size: 5 LOC McCabe index: 2 number of parameters: 3 id: 562 unit: def multiplicative_jitter() file: mesh_tensorflow/layers.py start line: 0 end line: 0 size: 5 LOC McCabe index: 2 number of parameters: 2 id: 563 unit: def get_tensor_size() file: mesh_tensorflow/auto_mtf/graph_interface.py start line: 0 end line: 0 size: 5 LOC McCabe index: 1 number of parameters: 4 id: 564 unit: def _initialize_operation_name_to_id() file: mesh_tensorflow/auto_mtf/graph_interface.py start line: 0 end line: 0 size: 5 LOC McCabe index: 2 number of parameters: 1 id: 565 unit: def _build_objective_function() file: mesh_tensorflow/auto_mtf/layout_optimizer.py start line: 0 end line: 0 size: 5 LOC McCabe index: 1 number of parameters: 1 id: 566 unit: def _initialize_mesh_dimension_name_to_size() file: mesh_tensorflow/auto_mtf/valid_layouts.py start line: 0 end line: 0 size: 5 LOC McCabe index: 2 number of parameters: 2 id: 567 unit: def _reshape_to_cubes() file: mesh_tensorflow/experimental/unet.py start line: 0 end line: 0 size: 5 LOC McCabe index: 1 number of parameters: 2 id: 568 unit: def alltoall() file: mesh_tensorflow/placement_mesh_impl.py start line: 0 end line: 0 size: 5 LOC McCabe index: 1 number of parameters: 5 id: 569 unit: def cumprod_to_tensor_axis() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 5 LOC McCabe index: 2 number of parameters: 2 id: 570 unit: def mesh_axis_to_tensor_axis() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 5 LOC McCabe index: 3 number of parameters: 2 id: 571 unit: def variable_placer_fn() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 5 LOC McCabe index: 2 number of parameters: 1 id: 572 unit: def mesh_axis_to_cumprod() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 5 LOC McCabe index: 3 number of parameters: 2 id: 573 unit: def to_string() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 5 LOC McCabe index: 3 number of parameters: 1 id: 574 unit: def sigmoid() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 5 LOC McCabe index: 1 number of parameters: 2 id: 575 unit: def tanh() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 5 LOC McCabe index: 1 number of parameters: 2 id: 576 unit: def __init__() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 5 LOC McCabe index: 2 number of parameters: 4 id: 577 unit: def lower() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 5 LOC McCabe index: 1 number of parameters: 2 id: 578 unit: def lower() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 5 LOC McCabe index: 1 number of parameters: 2 id: 579 unit: def lower() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 5 LOC McCabe index: 1 number of parameters: 2 id: 580 unit: def broadcast() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 5 LOC McCabe index: 2 number of parameters: 2 id: 581 unit: def apply_grad() file: mesh_tensorflow/optimize.py start line: 0 end line: 0 size: 5 LOC McCabe index: 3 number of parameters: 3 id: 582 unit: def _parameter_scale() file: mesh_tensorflow/optimize.py start line: 0 end line: 0 size: 5 LOC McCabe index: 3 number of parameters: 2 id: 583 unit: def get_position() file: mesh_tensorflow/transformer/transformer.py start line: 0 end line: 0 size: 5 LOC McCabe index: 2 number of parameters: 1 id: 584 unit: def sublayer_call_layer() file: mesh_tensorflow/transformer/transformer.py start line: 0 end line: 0 size: 5 LOC McCabe index: 1 number of parameters: 3 id: 585 unit: def sublayer_true_layer_norm() file: mesh_tensorflow/transformer/transformer.py start line: 0 end line: 0 size: 5 LOC McCabe index: 1 number of parameters: 4 id: 586 unit: def init_checkpoint_variable_mapping() file: mesh_tensorflow/transformer/utils.py start line: 0 end line: 0 size: 5 LOC McCabe index: 2 number of parameters: 2 id: 587 unit: def should_load_variable() file: mesh_tensorflow/transformer/utils.py start line: 0 end line: 0 size: 5 LOC McCabe index: 2 number of parameters: 2 id: 588 unit: def get_step_from_checkpoint_path() file: mesh_tensorflow/transformer/utils.py start line: 0 end line: 0 size: 5 LOC McCabe index: 2 number of parameters: 1 id: 589 unit: def clean_decodes() file: mesh_tensorflow/transformer/utils.py start line: 0 end line: 0 size: 5 LOC McCabe index: 1 number of parameters: 4 id: 590 unit: def attention_fn() file: mesh_tensorflow/transformer/transformer_layers.py start line: 0 end line: 0 size: 5 LOC McCabe index: 2 number of parameters: 1 id: 591 unit: def call() file: mesh_tensorflow/transformer/transformer_layers.py start line: 0 end line: 0 size: 5 LOC McCabe index: 1 number of parameters: 4 id: 592 unit: def __init__() file: mesh_tensorflow/transformer/transformer_layers.py start line: 0 end line: 0 size: 5 LOC McCabe index: 2 number of parameters: 3 id: 593 unit: def hidden_to_logits() file: mesh_tensorflow/transformer/vocab_embeddings.py start line: 0 end line: 0 size: 5 LOC McCabe index: 1 number of parameters: 3 id: 594 unit: def __init__() file: mesh_tensorflow/transformer/evolved_transformer.py start line: 0 end line: 0 size: 5 LOC McCabe index: 1 number of parameters: 2 id: 595 unit: def handle() file: mesh_tensorflow/tpu_variables.py start line: 0 end line: 0 size: 5 LOC McCabe index: 2 number of parameters: 1 id: 596 unit: def _read_variable_op() file: mesh_tensorflow/tpu_variables.py start line: 0 end line: 0 size: 5 LOC McCabe index: 2 number of parameters: 1 id: 597 unit: def __init__() file: mesh_tensorflow/bert/run_squad.py start line: 0 end line: 0 size: 5 LOC McCabe index: 1 number of parameters: 3 id: 598 unit: def __init__() file: mesh_tensorflow/bert/run_classifier.py start line: 0 end line: 0 size: 5 LOC McCabe index: 1 number of parameters: 5 id: 599 unit: def convert_by_vocab() file: mesh_tensorflow/bert/tokenization.py start line: 0 end line: 0 size: 5 LOC McCabe index: 2 number of parameters: 2 id: 600 unit: def weights_nonzero() file: mesh_tensorflow/layers.py start line: 0 end line: 0 size: 4 LOC McCabe index: 1 number of parameters: 2 id: 601 unit: def embedding() file: mesh_tensorflow/layers.py start line: 0 end line: 0 size: 4 LOC McCabe index: 1 number of parameters: 5 id: 602 unit: def _get_tensors() file: mesh_tensorflow/auto_mtf/graph_interface.py start line: 0 end line: 0 size: 4 LOC McCabe index: 3 number of parameters: 1 id: 603 unit: def get_layout_validator() file: mesh_tensorflow/auto_mtf/memory_estimator.py start line: 0 end line: 0 size: 4 LOC McCabe index: 2 number of parameters: 1 id: 604 unit: def get_graph_interface() file: mesh_tensorflow/auto_mtf/memory_estimator.py start line: 0 end line: 0 size: 4 LOC McCabe index: 2 number of parameters: 1 id: 605 unit: def _compute_graph_interface() file: mesh_tensorflow/auto_mtf/memory_estimator.py start line: 0 end line: 0 size: 4 LOC McCabe index: 2 number of parameters: 1 id: 606 unit: def is_valid_assignment() file: mesh_tensorflow/auto_mtf/valid_layouts.py start line: 0 end line: 0 size: 4 LOC McCabe index: 2 number of parameters: 3 id: 607 unit: def __init__() file: mesh_tensorflow/experimental/unet.py start line: 0 end line: 0 size: 4 LOC McCabe index: 1 number of parameters: 1 id: 608 unit: def _truncated_normal() file: mesh_tensorflow/experimental/data_aug_lib.py start line: 0 end line: 0 size: 4 LOC McCabe index: 1 number of parameters: 2 id: 609 unit: def __init__() file: mesh_tensorflow/placement_mesh_impl.py start line: 0 end line: 0 size: 4 LOC McCabe index: 1 number of parameters: 4 id: 610 unit: def assign_to_slices() file: mesh_tensorflow/placement_mesh_impl.py start line: 0 end line: 0 size: 4 LOC McCabe index: 1 number of parameters: 3 id: 611 unit: def allreduce() file: mesh_tensorflow/placement_mesh_impl.py start line: 0 end line: 0 size: 4 LOC McCabe index: 1 number of parameters: 4 id: 612 unit: def allconcat() file: mesh_tensorflow/placement_mesh_impl.py start line: 0 end line: 0 size: 4 LOC McCabe index: 1 number of parameters: 4 id: 613 unit: def __init__() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 4 LOC McCabe index: 3 number of parameters: 2 id: 614 unit: def capture_operations() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 4 LOC McCabe index: 1 number of parameters: 2 id: 615 unit: def mesh_impl() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 4 LOC McCabe index: 2 number of parameters: 2 id: 616 unit: def export_to_tf_tensor() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 4 LOC McCabe index: 1 number of parameters: 2 id: 617 unit: def __init__() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 4 LOC McCabe index: 1 number of parameters: 4 id: 618 unit: def tensor_layout() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 4 LOC McCabe index: 2 number of parameters: 2 id: 619 unit: def Print() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 4 LOC McCabe index: 1 number of parameters: 5 id: 620 unit: def has_gradient() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 4 LOC McCabe index: 6 number of parameters: 1 id: 621 unit: def has_gradient() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 4 LOC McCabe index: 2 number of parameters: 1 id: 622 unit: def gradient() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 4 LOC McCabe index: 2 number of parameters: 2 id: 623 unit: def square() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 4 LOC McCabe index: 1 number of parameters: 2 id: 624 unit: def sqrt() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 4 LOC McCabe index: 1 number of parameters: 2 id: 625 unit: def log() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 4 LOC McCabe index: 1 number of parameters: 2 id: 626 unit: def reciprocal() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 4 LOC McCabe index: 1 number of parameters: 2 id: 627 unit: def __init__() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 4 LOC McCabe index: 2 number of parameters: 4 id: 628 unit: def floordiv() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 4 LOC McCabe index: 2 number of parameters: 3 id: 629 unit: def mod() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 4 LOC McCabe index: 2 number of parameters: 3 id: 630 unit: def gradient() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 4 LOC McCabe index: 1 number of parameters: 2 id: 631 unit: def gradient() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 4 LOC McCabe index: 1 number of parameters: 2 id: 632 unit: def __init__() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 4 LOC McCabe index: 2 number of parameters: 5 id: 633 unit: def pool_backprop() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 4 LOC McCabe index: 1 number of parameters: 7 id: 634 unit: def adafactor_decay_rate_adam() file: mesh_tensorflow/optimize.py start line: 0 end line: 0 size: 4 LOC McCabe index: 1 number of parameters: 1 id: 635 unit: def encode_dataset() file: mesh_tensorflow/transformer/dataset.py start line: 0 end line: 0 size: 4 LOC McCabe index: 2 number of parameters: 2 id: 636 unit: def pad_dataset_with_zeroed_out_examples() file: mesh_tensorflow/transformer/dataset.py start line: 0 end line: 0 size: 4 LOC McCabe index: 2 number of parameters: 1 id: 637 unit: def encoder_length_dim() file: mesh_tensorflow/transformer/transformer.py start line: 0 end line: 0 size: 4 LOC McCabe index: 3 number of parameters: 1 id: 638 unit: def get_constant_state() file: mesh_tensorflow/transformer/transformer.py start line: 0 end line: 0 size: 4 LOC McCabe index: 1 number of parameters: 1 id: 639 unit: def _call_sublayers() file: mesh_tensorflow/transformer/transformer.py start line: 0 end line: 0 size: 4 LOC McCabe index: 2 number of parameters: 4 id: 640 unit: def __init__() file: mesh_tensorflow/transformer/transformer.py start line: 0 end line: 0 size: 4 LOC McCabe index: 1 number of parameters: 4 id: 641 unit: def delimited_lm_inputs_mask() file: mesh_tensorflow/transformer/transformer.py start line: 0 end line: 0 size: 4 LOC McCabe index: 1 number of parameters: 2 id: 642 unit: def reduce_ensemble_logits_mean_prob() file: mesh_tensorflow/transformer/transformer.py start line: 0 end line: 0 size: 4 LOC McCabe index: 1 number of parameters: 3 id: 643 unit: def get_cluster_mask() file: mesh_tensorflow/transformer/adaptive_softmax.py start line: 0 end line: 0 size: 4 LOC McCabe index: 1 number of parameters: 2 id: 644 unit: def ids_to_embedding() file: mesh_tensorflow/transformer/adaptive_softmax.py start line: 0 end line: 0 size: 4 LOC McCabe index: 2 number of parameters: 3 id: 645 unit: def inputs_vocabulary() file: mesh_tensorflow/transformer/utils.py start line: 0 end line: 0 size: 4 LOC McCabe index: 2 number of parameters: 1 id: 646 unit: def targets_vocabulary() file: mesh_tensorflow/transformer/utils.py start line: 0 end line: 0 size: 4 LOC McCabe index: 2 number of parameters: 1 id: 647 unit: def auto_train_steps() file: mesh_tensorflow/transformer/utils.py start line: 0 end line: 0 size: 4 LOC McCabe index: 1 number of parameters: 3 id: 648 unit: def expected_attention_output_shape() file: mesh_tensorflow/transformer/transformer_layers.py start line: 0 end line: 0 size: 4 LOC McCabe index: 2 number of parameters: 3 id: 649 unit: def _get_memory_antecedent() file: mesh_tensorflow/transformer/transformer_layers.py start line: 0 end line: 0 size: 4 LOC McCabe index: 1 number of parameters: 2 id: 650 unit: def __init__() file: mesh_tensorflow/transformer/transformer_layers.py start line: 0 end line: 0 size: 4 LOC McCabe index: 1 number of parameters: 4 id: 651 unit: def ids_to_embedding() file: mesh_tensorflow/transformer/vocab_embeddings.py start line: 0 end line: 0 size: 4 LOC McCabe index: 1 number of parameters: 3 id: 652 unit: def __init__() file: mesh_tensorflow/transformer/vocab_embeddings.py start line: 0 end line: 0 size: 4 LOC McCabe index: 1 number of parameters: 4 id: 653 unit: def ids_to_embedding() file: mesh_tensorflow/transformer/vocab_embeddings.py start line: 0 end line: 0 size: 4 LOC McCabe index: 2 number of parameters: 3 id: 654 unit: def _call_sublayers() file: mesh_tensorflow/transformer/funnel_transformer.py start line: 0 end line: 0 size: 4 LOC McCabe index: 2 number of parameters: 4 id: 655 unit: def from_json_file() file: mesh_tensorflow/bert/bert.py start line: 0 end line: 0 size: 4 LOC McCabe index: 1 number of parameters: 2 id: 656 unit: def normalize() file: mesh_tensorflow/bert/bert.py start line: 0 end line: 0 size: 4 LOC McCabe index: 1 number of parameters: 2 id: 657 unit: def key_dim() file: mesh_tensorflow/bert/bert.py start line: 0 end line: 0 size: 4 LOC McCabe index: 2 number of parameters: 1 id: 658 unit: def key_heads_dims() file: mesh_tensorflow/bert/bert.py start line: 0 end line: 0 size: 4 LOC McCabe index: 2 number of parameters: 1 id: 659 unit: def value_dim() file: mesh_tensorflow/bert/bert.py start line: 0 end line: 0 size: 4 LOC McCabe index: 2 number of parameters: 1 id: 660 unit: def value_heads_dims() file: mesh_tensorflow/bert/bert.py start line: 0 end line: 0 size: 4 LOC McCabe index: 2 number of parameters: 1 id: 661 unit: def get_dev_examples() file: mesh_tensorflow/bert/run_classifier.py start line: 0 end line: 0 size: 4 LOC McCabe index: 1 number of parameters: 2 id: 662 unit: def __init__() file: mesh_tensorflow/bert/tokenization.py start line: 0 end line: 0 size: 4 LOC McCabe index: 1 number of parameters: 4 id: 663 unit: def rename_length_to_memory_length() file: mesh_tensorflow/layers.py start line: 0 end line: 0 size: 3 LOC McCabe index: 1 number of parameters: 3 id: 664 unit: def attention_mask_ignore_padding() file: mesh_tensorflow/layers.py start line: 0 end line: 0 size: 3 LOC McCabe index: 1 number of parameters: 2 id: 665 unit: def attention_mask_autoregressive() file: mesh_tensorflow/layers.py start line: 0 end line: 0 size: 3 LOC McCabe index: 1 number of parameters: 2 id: 666 unit: def max_pool2d() file: mesh_tensorflow/layers.py start line: 0 end line: 0 size: 3 LOC McCabe index: 2 number of parameters: 3 id: 667 unit: def max_pool3d() file: mesh_tensorflow/layers.py start line: 0 end line: 0 size: 3 LOC McCabe index: 2 number of parameters: 4 id: 668 unit: def avg_pool2d() file: mesh_tensorflow/layers.py start line: 0 end line: 0 size: 3 LOC McCabe index: 2 number of parameters: 3 id: 669 unit: def avg_pool3d() file: mesh_tensorflow/layers.py start line: 0 end line: 0 size: 3 LOC McCabe index: 2 number of parameters: 4 id: 670 unit: def __init__() file: mesh_tensorflow/simd_mesh_impl.py start line: 0 end line: 0 size: 3 LOC McCabe index: 1 number of parameters: 2 id: 671 unit: def get_all_operation_names() file: mesh_tensorflow/auto_mtf/graph_interface.py start line: 0 end line: 0 size: 3 LOC McCabe index: 2 number of parameters: 1 id: 672 unit: def get_operation_input_names() file: mesh_tensorflow/auto_mtf/graph_interface.py start line: 0 end line: 0 size: 3 LOC McCabe index: 2 number of parameters: 2 id: 673 unit: def get_operation_output_names() file: mesh_tensorflow/auto_mtf/graph_interface.py start line: 0 end line: 0 size: 3 LOC McCabe index: 2 number of parameters: 2 id: 674 unit: def get_all_tensor_names() file: mesh_tensorflow/auto_mtf/graph_interface.py start line: 0 end line: 0 size: 3 LOC McCabe index: 2 number of parameters: 1 id: 675 unit: def is_tensor_on_canonical_device() file: mesh_tensorflow/auto_mtf/graph_interface.py start line: 0 end line: 0 size: 3 LOC McCabe index: 2 number of parameters: 2 id: 676 unit: def set_tensor_final() file: mesh_tensorflow/auto_mtf/graph_interface.py start line: 0 end line: 0 size: 3 LOC McCabe index: 1 number of parameters: 2 id: 677 unit: def is_tensor_final() file: mesh_tensorflow/auto_mtf/graph_interface.py start line: 0 end line: 0 size: 3 LOC McCabe index: 1 number of parameters: 2 id: 678 unit: def _name_to_tensor() file: mesh_tensorflow/auto_mtf/graph_interface.py start line: 0 end line: 0 size: 3 LOC McCabe index: 1 number of parameters: 2 id: 679 unit: def _compute_layout_validator() file: mesh_tensorflow/auto_mtf/memory_estimator.py start line: 0 end line: 0 size: 3 LOC McCabe index: 1 number of parameters: 1 id: 680 unit: def main() file: mesh_tensorflow/experimental/offline_data_aug.py start line: 0 end line: 0 size: 3 LOC McCabe index: 1 number of parameters: 1 id: 681 unit: def __init__() file: mesh_tensorflow/experimental/model_executor.py start line: 0 end line: 0 size: 3 LOC McCabe index: 1 number of parameters: 1 id: 682 unit: def outside_all_rewrites() file: mesh_tensorflow/utils.py start line: 0 end line: 0 size: 3 LOC McCabe index: 1 number of parameters: 0 id: 683 unit: def to_string() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 3 LOC McCabe index: 2 number of parameters: 1 id: 684 unit: def copy_slices_to_masters() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 3 LOC McCabe index: 2 number of parameters: 1 id: 685 unit: def add_counter() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 3 LOC McCabe index: 1 number of parameters: 3 id: 686 unit: def set_tensor_lowering() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 3 LOC McCabe index: 1 number of parameters: 3 id: 687 unit: def __init__() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 3 LOC McCabe index: 1 number of parameters: 3 id: 688 unit: def tensor_dimension_to_mesh_axis() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 3 LOC McCabe index: 1 number of parameters: 2 id: 689 unit: def splittable_dims() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 3 LOC McCabe index: 1 number of parameters: 1 id: 690 unit: def unsplittable_dims() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 3 LOC McCabe index: 1 number of parameters: 1 id: 691 unit: def _initialize_all_dimensions_as_splittable() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 3 LOC McCabe index: 1 number of parameters: 1 id: 692 unit: def rsqrt() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 3 LOC McCabe index: 1 number of parameters: 2 id: 693 unit: def exp() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 3 LOC McCabe index: 1 number of parameters: 2 id: 694 unit: def mtf_pow() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 3 LOC McCabe index: 1 number of parameters: 2 id: 695 unit: def negative() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 3 LOC McCabe index: 1 number of parameters: 2 id: 696 unit: def swish() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 3 LOC McCabe index: 1 number of parameters: 1 id: 697 unit: def softplus() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 3 LOC McCabe index: 1 number of parameters: 1 id: 698 unit: def squared_relu() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 3 LOC McCabe index: 1 number of parameters: 1 id: 699 unit: def sign() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 3 LOC McCabe index: 1 number of parameters: 2 id: 700 unit: def mtf_abs() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 3 LOC McCabe index: 1 number of parameters: 1 id: 701 unit: def gradient() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 3 LOC McCabe index: 1 number of parameters: 2 id: 702 unit: def less() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 3 LOC McCabe index: 1 number of parameters: 3 id: 703 unit: def greater() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 3 LOC McCabe index: 1 number of parameters: 3 id: 704 unit: def less_equal() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 3 LOC McCabe index: 1 number of parameters: 3 id: 705 unit: def greater_equal() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 3 LOC McCabe index: 1 number of parameters: 3 id: 706 unit: def equal() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 3 LOC McCabe index: 1 number of parameters: 3 id: 707 unit: def not_equal() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 3 LOC McCabe index: 1 number of parameters: 3 id: 708 unit: def logical_and() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 3 LOC McCabe index: 1 number of parameters: 3 id: 709 unit: def logical_or() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 3 LOC McCabe index: 1 number of parameters: 3 id: 710 unit: def __init__() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 3 LOC McCabe index: 2 number of parameters: 5 id: 711 unit: def __init__() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 3 LOC McCabe index: 2 number of parameters: 6 id: 712 unit: def gradient() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 3 LOC McCabe index: 1 number of parameters: 2 id: 713 unit: def __init__() file: mesh_tensorflow/optimize.py start line: 0 end line: 0 size: 3 LOC McCabe index: 1 number of parameters: 3 id: 714 unit: def adafactor_decay_rate_pow() file: mesh_tensorflow/optimize.py start line: 0 end line: 0 size: 3 LOC McCabe index: 1 number of parameters: 2 id: 715 unit: def encode_tf() file: mesh_tensorflow/transformer/t2t_vocabulary.py start line: 0 end line: 0 size: 3 LOC McCabe index: 1 number of parameters: 2 id: 716 unit: def get_t2t_vocabulary() file: mesh_tensorflow/transformer/t2t_vocabulary.py start line: 0 end line: 0 size: 3 LOC McCabe index: 1 number of parameters: 2 id: 717 unit: def simple_text_line_dataset() file: mesh_tensorflow/transformer/dataset.py start line: 0 end line: 0 size: 3 LOC McCabe index: 1 number of parameters: 2 id: 718 unit: def sublayer_residual() file: mesh_tensorflow/transformer/transformer.py start line: 0 end line: 0 size: 3 LOC McCabe index: 1 number of parameters: 3 id: 719 unit: def sublayer_clip_activation_gradient() file: mesh_tensorflow/transformer/transformer.py start line: 0 end line: 0 size: 3 LOC McCabe index: 1 number of parameters: 4 id: 720 unit: def sublayer_legacy_dropout() file: mesh_tensorflow/transformer/transformer.py start line: 0 end line: 0 size: 3 LOC McCabe index: 1 number of parameters: 3 id: 721 unit: def reduce_ensemble_logits_select() file: mesh_tensorflow/transformer/transformer.py start line: 0 end line: 0 size: 3 LOC McCabe index: 1 number of parameters: 4 id: 722 unit: def reduce_ensemble_logits_mean_logit() file: mesh_tensorflow/transformer/transformer.py start line: 0 end line: 0 size: 3 LOC McCabe index: 1 number of parameters: 3 id: 723 unit: def reduce_ensemble_logits() file: mesh_tensorflow/transformer/transformer.py start line: 0 end line: 0 size: 3 LOC McCabe index: 1 number of parameters: 4 id: 724 unit: def get_tfds_vocabulary() file: mesh_tensorflow/transformer/vocabulary.py start line: 0 end line: 0 size: 3 LOC McCabe index: 3 number of parameters: 1 id: 725 unit: def get_cluster_length_dim() file: mesh_tensorflow/transformer/adaptive_softmax.py start line: 0 end line: 0 size: 3 LOC McCabe index: 1 number of parameters: 2 id: 726 unit: def compute_log_softmax() file: mesh_tensorflow/transformer/adaptive_softmax.py start line: 0 end line: 0 size: 3 LOC McCabe index: 1 number of parameters: 3 id: 727 unit: def __init__() file: mesh_tensorflow/transformer/moe.py start line: 0 end line: 0 size: 3 LOC McCabe index: 2 number of parameters: 2 id: 728 unit: def layer_output_from_attention_output() file: mesh_tensorflow/transformer/transformer_layers.py start line: 0 end line: 0 size: 3 LOC McCabe index: 1 number of parameters: 4 id: 729 unit: def rename_length_to_memory_length() file: mesh_tensorflow/transformer/transformer_layers.py start line: 0 end line: 0 size: 3 LOC McCabe index: 1 number of parameters: 3 id: 730 unit: def __init__() file: mesh_tensorflow/transformer/transformer_layers.py start line: 0 end line: 0 size: 3 LOC McCabe index: 1 number of parameters: 3 id: 731 unit: def call() file: mesh_tensorflow/transformer/transformer_layers.py start line: 0 end line: 0 size: 3 LOC McCabe index: 1 number of parameters: 4 id: 732 unit: def __init__() file: mesh_tensorflow/transformer/transformer_layers.py start line: 0 end line: 0 size: 3 LOC McCabe index: 1 number of parameters: 3 id: 733 unit: def __init__() file: mesh_tensorflow/transformer/transformer_layers.py start line: 0 end line: 0 size: 3 LOC McCabe index: 1 number of parameters: 3 id: 734 unit: def __init__() file: mesh_tensorflow/transformer/transformer_layers.py start line: 0 end line: 0 size: 3 LOC McCabe index: 1 number of parameters: 3 id: 735 unit: def _constraint() file: mesh_tensorflow/transformer/transformer_layers.py start line: 0 end line: 0 size: 3 LOC McCabe index: 1 number of parameters: 2 id: 736 unit: def __init__() file: mesh_tensorflow/transformer/transformer_layers.py start line: 0 end line: 0 size: 3 LOC McCabe index: 1 number of parameters: 3 id: 737 unit: def call() file: mesh_tensorflow/transformer/transformer_layers.py start line: 0 end line: 0 size: 3 LOC McCabe index: 1 number of parameters: 4 id: 738 unit: def ids_to_embedding() file: mesh_tensorflow/transformer/vocab_embeddings.py start line: 0 end line: 0 size: 3 LOC McCabe index: 2 number of parameters: 3 id: 739 unit: def ids_to_embedding() file: mesh_tensorflow/transformer/vocab_embeddings.py start line: 0 end line: 0 size: 3 LOC McCabe index: 1 number of parameters: 3 id: 740 unit: def ids_to_embedding() file: mesh_tensorflow/transformer/vocab_embeddings.py start line: 0 end line: 0 size: 3 LOC McCabe index: 1 number of parameters: 3 id: 741 unit: def __init__() file: mesh_tensorflow/transformer/fixup_layers.py start line: 0 end line: 0 size: 3 LOC McCabe index: 1 number of parameters: 3 id: 742 unit: def call() file: mesh_tensorflow/transformer/fixup_layers.py start line: 0 end line: 0 size: 3 LOC McCabe index: 1 number of parameters: 4 id: 743 unit: def constant() file: mesh_tensorflow/transformer/learning_rate_schedules.py start line: 0 end line: 0 size: 3 LOC McCabe index: 1 number of parameters: 3 id: 744 unit: def constant_learning_rate() file: mesh_tensorflow/transformer/learning_rate_schedules.py start line: 0 end line: 0 size: 3 LOC McCabe index: 1 number of parameters: 3 id: 745 unit: def compute_q() file: mesh_tensorflow/transformer/attention.py start line: 0 end line: 0 size: 3 LOC McCabe index: 1 number of parameters: 2 id: 746 unit: def compute_k() file: mesh_tensorflow/transformer/attention.py start line: 0 end line: 0 size: 3 LOC McCabe index: 1 number of parameters: 2 id: 747 unit: def compute_kv() file: mesh_tensorflow/transformer/attention.py start line: 0 end line: 0 size: 3 LOC McCabe index: 1 number of parameters: 2 id: 748 unit: def compute_v() file: mesh_tensorflow/transformer/attention.py start line: 0 end line: 0 size: 3 LOC McCabe index: 1 number of parameters: 2 id: 749 unit: def _handle_graph() file: mesh_tensorflow/tpu_variables.py start line: 0 end line: 0 size: 3 LOC McCabe index: 1 number of parameters: 1 id: 750 unit: def to_dict() file: mesh_tensorflow/bert/bert.py start line: 0 end line: 0 size: 3 LOC McCabe index: 1 number of parameters: 1 id: 751 unit: def vocab_dim() file: mesh_tensorflow/bert/bert.py start line: 0 end line: 0 size: 3 LOC McCabe index: 1 number of parameters: 1 id: 752 unit: def feedforward_intermediate_dim() file: mesh_tensorflow/bert/bert.py start line: 0 end line: 0 size: 3 LOC McCabe index: 1 number of parameters: 1 id: 753 unit: def max_position_embeddings_dim() file: mesh_tensorflow/bert/bert.py start line: 0 end line: 0 size: 3 LOC McCabe index: 1 number of parameters: 1 id: 754 unit: def get_train_examples() file: mesh_tensorflow/bert/run_classifier.py start line: 0 end line: 0 size: 3 LOC McCabe index: 1 number of parameters: 2 id: 755 unit: def get_test_examples() file: mesh_tensorflow/bert/run_classifier.py start line: 0 end line: 0 size: 3 LOC McCabe index: 1 number of parameters: 2 id: 756 unit: def get_train_examples() file: mesh_tensorflow/bert/run_classifier.py start line: 0 end line: 0 size: 3 LOC McCabe index: 1 number of parameters: 2 id: 757 unit: def get_dev_examples() file: mesh_tensorflow/bert/run_classifier.py start line: 0 end line: 0 size: 3 LOC McCabe index: 1 number of parameters: 2 id: 758 unit: def get_test_examples() file: mesh_tensorflow/bert/run_classifier.py start line: 0 end line: 0 size: 3 LOC McCabe index: 1 number of parameters: 2 id: 759 unit: def get_train_examples() file: mesh_tensorflow/bert/run_classifier.py start line: 0 end line: 0 size: 3 LOC McCabe index: 1 number of parameters: 2 id: 760 unit: def get_dev_examples() file: mesh_tensorflow/bert/run_classifier.py start line: 0 end line: 0 size: 3 LOC McCabe index: 1 number of parameters: 2 id: 761 unit: def get_test_examples() file: mesh_tensorflow/bert/run_classifier.py start line: 0 end line: 0 size: 3 LOC McCabe index: 1 number of parameters: 2 id: 762 unit: def __init__() file: mesh_tensorflow/bert/tokenization.py start line: 0 end line: 0 size: 3 LOC McCabe index: 1 number of parameters: 3 id: 763 unit: def unit_scaling_convention() file: mesh_tensorflow/layers.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 764 unit: def __call__() file: mesh_tensorflow/layers.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 3 id: 765 unit: def _half_residual_and_swap() file: mesh_tensorflow/layers.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 5 id: 766 unit: def l2p() file: mesh_tensorflow/simd_mesh_impl.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 2 id: 767 unit: def p2l() file: mesh_tensorflow/simd_mesh_impl.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 2 id: 768 unit: def __repr__() file: mesh_tensorflow/simd_mesh_impl.py start line: 0 end line: 0 size: 2 LOC McCabe index: 2 number of parameters: 1 id: 769 unit: def tensor_list() file: mesh_tensorflow/simd_mesh_impl.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 770 unit: def one_slice() file: mesh_tensorflow/simd_mesh_impl.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 771 unit: def from_tensor_list() file: mesh_tensorflow/simd_mesh_impl.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 2 id: 772 unit: def all_slices() file: mesh_tensorflow/simd_mesh_impl.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 773 unit: def slice_shape() file: mesh_tensorflow/simd_mesh_impl.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 774 unit: def to_laid_out_tensor() file: mesh_tensorflow/simd_mesh_impl.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 775 unit: def laid_out_tensor() file: mesh_tensorflow/simd_mesh_impl.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 776 unit: def copy_master_to_slices() file: mesh_tensorflow/simd_mesh_impl.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 777 unit: def copy_slices_to_master() file: mesh_tensorflow/simd_mesh_impl.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 778 unit: def laid_out_pnum() file: mesh_tensorflow/simd_mesh_impl.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 779 unit: def slice() file: mesh_tensorflow/simd_mesh_impl.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 3 id: 780 unit: def get_num_operations() file: mesh_tensorflow/auto_mtf/graph_interface.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 781 unit: def get_tensor_dtype() file: mesh_tensorflow/auto_mtf/graph_interface.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 2 id: 782 unit: def _name_to_operation() file: mesh_tensorflow/auto_mtf/graph_interface.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 2 id: 783 unit: def _minimize_peak_memory_naive() file: mesh_tensorflow/auto_mtf/scheduler.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 784 unit: def _global_var_name() file: mesh_tensorflow/auto_mtf/layout_optimizer.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 2 id: 785 unit: def splittable_mtf_dimension_names() file: mesh_tensorflow/auto_mtf/valid_layouts.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 786 unit: def mesh_dimension_name_to_size() file: mesh_tensorflow/auto_mtf/valid_layouts.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 787 unit: def initializer() file: mesh_tensorflow/experimental/input_reader.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 788 unit: def _init_slice_cache() file: mesh_tensorflow/experimental/input_reader.py start line: 0 end line: 0 size: 2 LOC McCabe index: 2 number of parameters: 1 id: 789 unit: def ordered_ordinals() file: mesh_tensorflow/experimental/input_reader.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 790 unit: def ordered_tpus() file: mesh_tensorflow/experimental/input_reader.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 791 unit: def ordered_hosts() file: mesh_tensorflow/experimental/input_reader.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 792 unit: def ordered_host_ids() file: mesh_tensorflow/experimental/input_reader.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 793 unit: def host_id_to_its_pnums() file: mesh_tensorflow/experimental/input_reader.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 794 unit: def num_cores() file: mesh_tensorflow/experimental/input_reader.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 795 unit: def num_hosts() file: mesh_tensorflow/experimental/input_reader.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 796 unit: def num_cores_per_host() file: mesh_tensorflow/experimental/input_reader.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 797 unit: def infeed_queue() file: mesh_tensorflow/experimental/input_reader.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 798 unit: def initialize() file: mesh_tensorflow/experimental/input_reader.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 2 id: 799 unit: def device_assignment() file: mesh_tensorflow/experimental/model_executor.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 800 unit: def num_hosts() file: mesh_tensorflow/experimental/model_executor.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 801 unit: def num_cores() file: mesh_tensorflow/experimental/model_executor.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 802 unit: def num_cores_per_host() file: mesh_tensorflow/experimental/model_executor.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 803 unit: def mesh_impl() file: mesh_tensorflow/experimental/model_executor.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 804 unit: def main() file: mesh_tensorflow/experimental/model_executor.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 805 unit: def get_layout() file: mesh_tensorflow/experimental/unet.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 0 id: 806 unit: def __init__() file: mesh_tensorflow/placement_mesh_impl.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 2 id: 807 unit: def __repr__() file: mesh_tensorflow/placement_mesh_impl.py start line: 0 end line: 0 size: 2 LOC McCabe index: 2 number of parameters: 1 id: 808 unit: def tensor_list() file: mesh_tensorflow/placement_mesh_impl.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 809 unit: def from_tensor_list() file: mesh_tensorflow/placement_mesh_impl.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 2 id: 810 unit: def all_slices() file: mesh_tensorflow/placement_mesh_impl.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 811 unit: def slice_shape() file: mesh_tensorflow/placement_mesh_impl.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 812 unit: def to_laid_out_tensor() file: mesh_tensorflow/placement_mesh_impl.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 813 unit: def laid_out_tensor() file: mesh_tensorflow/placement_mesh_impl.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 814 unit: def copy_master_to_slices() file: mesh_tensorflow/placement_mesh_impl.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 815 unit: def copy_slices_to_master() file: mesh_tensorflow/placement_mesh_impl.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 816 unit: def laid_out_pnum() file: mesh_tensorflow/placement_mesh_impl.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 817 unit: def devices() file: mesh_tensorflow/placement_mesh_impl.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 818 unit: def export_to_tf_tensor() file: mesh_tensorflow/placement_mesh_impl.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 3 id: 819 unit: def import_tf_tensor() file: mesh_tensorflow/placement_mesh_impl.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 3 id: 820 unit: def dims() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 821 unit: def ndims() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 822 unit: def __repr__() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 823 unit: def __eq__() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 2 id: 824 unit: def __ne__() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 2 id: 825 unit: def __len__() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 826 unit: def __getitem__() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 2 id: 827 unit: def __iter__() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 828 unit: def to_integer_list() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 2 LOC McCabe index: 2 number of parameters: 1 id: 829 unit: def size() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 830 unit: def cumprod() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 831 unit: def dimension_names() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 2 LOC McCabe index: 2 number of parameters: 1 id: 832 unit: def __init__() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 2 id: 833 unit: def __repr__() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 834 unit: def mesh_dimension_name_to_tensor_dimension_names() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 2 LOC McCabe index: 3 number of parameters: 2 id: 835 unit: def __init__() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 2 id: 836 unit: def __eq__() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 2 id: 837 unit: def __ne__() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 2 id: 838 unit: def __repr__() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 839 unit: def __len__() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 840 unit: def __getitem__() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 2 id: 841 unit: def __iter__() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 842 unit: def tensor_axis_to_mesh_axis() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 843 unit: def is_fully_replicated() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 844 unit: def __repr__() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 845 unit: def operations() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 846 unit: def trainable_variables() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 847 unit: def all_variables() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 848 unit: def to_string() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 2 LOC McCabe index: 2 number of parameters: 1 id: 849 unit: def lowered_operation() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 2 id: 850 unit: def counters() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 851 unit: def laid_out_size() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 2 id: 852 unit: def graph() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 853 unit: def shape() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 854 unit: def ndims() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 855 unit: def layout_rules() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 856 unit: def size() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 857 unit: def supports_control_dependencies() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 858 unit: def slice_size() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 2 id: 859 unit: def laid_out_size() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 2 id: 860 unit: def slicewise() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 3 id: 861 unit: def allreduce() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 4 id: 862 unit: def allconcat() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 4 id: 863 unit: def alltoall() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 5 id: 864 unit: def receive() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 4 id: 865 unit: def laid_out_pnum() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 866 unit: def export_to_tf_tensor() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 3 id: 867 unit: def import_tf_tensor() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 3 id: 868 unit: def einsum() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 3 id: 869 unit: def slice_shape() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 870 unit: def shape() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 871 unit: def size() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 872 unit: def mesh() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 873 unit: def graph() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 874 unit: def operation() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 875 unit: def dtype() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 876 unit: def name() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 877 unit: def __repr__() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 878 unit: def __add__() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 2 id: 879 unit: def __radd__() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 2 id: 880 unit: def __sub__() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 2 id: 881 unit: def __rsub__() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 2 id: 882 unit: def __mul__() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 2 id: 883 unit: def __rmul__() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 2 id: 884 unit: def __neg__() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 885 unit: def __truediv__() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 2 id: 886 unit: def __rtruediv__() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 2 id: 887 unit: def __floordiv__() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 2 id: 888 unit: def __rfloordiv__() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 2 id: 889 unit: def __mod__() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 2 id: 890 unit: def __rmod__() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 2 id: 891 unit: def to_string() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 892 unit: def graph() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 893 unit: def mesh() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 894 unit: def name() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 895 unit: def inputs() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 896 unit: def outputs() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 897 unit: def gradient() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 2 id: 898 unit: def lower() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 2 id: 899 unit: def identity() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 2 id: 900 unit: def sin() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 2 id: 901 unit: def cos() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 2 id: 902 unit: def _rsqrt_grad() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 2 id: 903 unit: def logical_not() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 2 id: 904 unit: def _relu_grad() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 2 id: 905 unit: def relu() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 2 id: 906 unit: def to_float() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 2 id: 907 unit: def to_bfloat16() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 2 id: 908 unit: def to_int32() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 2 id: 909 unit: def gradient() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 2 id: 910 unit: def gradient() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 2 id: 911 unit: def gradient() file: mesh_tensorflow/ops.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 2 id: 912 unit: def apply_grad() file: mesh_tensorflow/optimize.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 3 id: 913 unit: def __init__() file: mesh_tensorflow/optimize.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 2 id: 914 unit: def lr() file: mesh_tensorflow/optimize.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 915 unit: def lr() file: mesh_tensorflow/optimize.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 916 unit: def momentum() file: mesh_tensorflow/optimize.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 917 unit: def _decay_rate_default() file: mesh_tensorflow/optimize.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 918 unit: def reduce_rms() file: mesh_tensorflow/optimize.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 2 id: 919 unit: def vocab_size() file: mesh_tensorflow/transformer/t2t_vocabulary.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 920 unit: def encode() file: mesh_tensorflow/transformer/t2t_vocabulary.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 2 id: 921 unit: def decode() file: mesh_tensorflow/transformer/t2t_vocabulary.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 2 id: 922 unit: def make_text_line_dataset() file: mesh_tensorflow/transformer/dataset.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 923 unit: def call() file: mesh_tensorflow/transformer/transformer.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 3 id: 924 unit: def to_json() file: mesh_tensorflow/transformer/transformer.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 925 unit: def set_name() file: mesh_tensorflow/transformer/transformer.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 2 id: 926 unit: def name() file: mesh_tensorflow/transformer/transformer.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 927 unit: def train() file: mesh_tensorflow/transformer/transformer.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 928 unit: def activation_dtype() file: mesh_tensorflow/transformer/transformer.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 929 unit: def get_states() file: mesh_tensorflow/transformer/transformer.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 2 id: 930 unit: def record_new_states() file: mesh_tensorflow/transformer/transformer.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 2 id: 931 unit: def record_constant_state() file: mesh_tensorflow/transformer/transformer.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 2 id: 932 unit: def num_layers() file: mesh_tensorflow/transformer/transformer.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 933 unit: def layers() file: mesh_tensorflow/transformer/transformer.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 934 unit: def sublayer_legacy_rms_norm() file: mesh_tensorflow/transformer/transformer.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 3 id: 935 unit: def sublayer_legacy_final_rms_norm() file: mesh_tensorflow/transformer/transformer.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 3 id: 936 unit: def fully_autoregressive() file: mesh_tensorflow/transformer/transformer.py start line: 0 end line: 0 size: 2 LOC McCabe index: 2 number of parameters: 1 id: 937 unit: def ensemble_dims() file: mesh_tensorflow/transformer/transformer.py start line: 0 end line: 0 size: 2 LOC McCabe index: 2 number of parameters: 1 id: 938 unit: def output_vocab_dim() file: mesh_tensorflow/transformer/transformer.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 939 unit: def loss_denominator() file: mesh_tensorflow/transformer/transformer.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 3 id: 940 unit: def z_loss() file: mesh_tensorflow/transformer/transformer.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 941 unit: def _round_up_to_multiple() file: mesh_tensorflow/transformer/transformer.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 2 id: 942 unit: def get_vocab_embedding_cls() file: mesh_tensorflow/transformer/transformer.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 943 unit: def vocab_size() file: mesh_tensorflow/transformer/vocabulary.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 944 unit: def encode() file: mesh_tensorflow/transformer/vocabulary.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 2 id: 945 unit: def decode() file: mesh_tensorflow/transformer/vocabulary.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 2 id: 946 unit: def encode_tf() file: mesh_tensorflow/transformer/vocabulary.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 2 id: 947 unit: def decode_tf() file: mesh_tensorflow/transformer/vocabulary.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 2 id: 948 unit: def __init__() file: mesh_tensorflow/transformer/vocabulary.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 2 id: 949 unit: def vocab_size() file: mesh_tensorflow/transformer/vocabulary.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 950 unit: def encode() file: mesh_tensorflow/transformer/vocabulary.py start line: 0 end line: 0 size: 2 LOC McCabe index: 3 number of parameters: 2 id: 951 unit: def decode() file: mesh_tensorflow/transformer/vocabulary.py start line: 0 end line: 0 size: 2 LOC McCabe index: 2 number of parameters: 2 id: 952 unit: def end_token_id() file: mesh_tensorflow/transformer/adaptive_softmax.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 953 unit: def length_projection_factor() file: mesh_tensorflow/transformer/adaptive_softmax.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 954 unit: def get_log_softmax_value() file: mesh_tensorflow/transformer/adaptive_softmax.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 3 id: 955 unit: def _layer_norm() file: mesh_tensorflow/transformer/universal_transformer.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 4 id: 956 unit: def num_layers() file: mesh_tensorflow/transformer/universal_transformer.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 957 unit: def layers() file: mesh_tensorflow/transformer/universal_transformer.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 958 unit: def add_hparam() file: mesh_tensorflow/transformer/moe.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 3 id: 959 unit: def _filter_features() file: mesh_tensorflow/transformer/utils.py start line: 0 end line: 0 size: 2 LOC McCabe index: 3 number of parameters: 1 id: 960 unit: def separate_vocabularies() file: mesh_tensorflow/transformer/utils.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 2 id: 961 unit: def variable_filter_max_size() file: mesh_tensorflow/transformer/utils.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 2 id: 962 unit: def kv_dim() file: mesh_tensorflow/transformer/transformer_layers.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 963 unit: def memory_length() file: mesh_tensorflow/transformer/transformer_layers.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 2 id: 964 unit: def min_relative_position() file: mesh_tensorflow/transformer/transformer_layers.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 2 id: 965 unit: def max_relative_position() file: mesh_tensorflow/transformer/transformer_layers.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 2 id: 966 unit: def _get_memory_antecedent() file: mesh_tensorflow/transformer/transformer_layers.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 2 id: 967 unit: def _get_memory_antecedent() file: mesh_tensorflow/transformer/transformer_layers.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 2 id: 968 unit: def attention_fn() file: mesh_tensorflow/transformer/transformer_layers.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 969 unit: def min_relative_position() file: mesh_tensorflow/transformer/transformer_layers.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 2 id: 970 unit: def max_relative_position() file: mesh_tensorflow/transformer/transformer_layers.py start line: 0 end line: 0 size: 2 LOC McCabe index: 2 number of parameters: 2 id: 971 unit: def window_dim() file: mesh_tensorflow/transformer/transformer_layers.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 972 unit: def _get_memory_antecedent() file: mesh_tensorflow/transformer/transformer_layers.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 2 id: 973 unit: def _get_memory_antecedent() file: mesh_tensorflow/transformer/transformer_layers.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 2 id: 974 unit: def _get_memory_antecedent() file: mesh_tensorflow/transformer/transformer_layers.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 2 id: 975 unit: def filter_size() file: mesh_tensorflow/transformer/transformer_layers.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 976 unit: def filter_size() file: mesh_tensorflow/transformer/transformer_layers.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 977 unit: def hidden_to_logits() file: mesh_tensorflow/transformer/vocab_embeddings.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 3 id: 978 unit: def _get_memory_antecedent() file: mesh_tensorflow/transformer/fixup_layers.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 2 id: 979 unit: def layers() file: mesh_tensorflow/transformer/funnel_transformer.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 980 unit: def set_context() file: mesh_tensorflow/transformer/funnel_transformer.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 2 id: 981 unit: def context() file: mesh_tensorflow/transformer/funnel_transformer.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 982 unit: def q_dims() file: mesh_tensorflow/transformer/attention.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 983 unit: def k_dims() file: mesh_tensorflow/transformer/attention.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 984 unit: def v_dims() file: mesh_tensorflow/transformer/attention.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 985 unit: def o_dims() file: mesh_tensorflow/transformer/attention.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 986 unit: def _combined_dim() file: mesh_tensorflow/transformer/attention.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 987 unit: def visibility_mask_to_attention_bias() file: mesh_tensorflow/transformer/attention.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 2 id: 988 unit: def initializer() file: mesh_tensorflow/tpu_variables.py start line: 0 end line: 0 size: 2 LOC McCabe index: 2 number of parameters: 1 id: 989 unit: def graph() file: mesh_tensorflow/tpu_variables.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 990 unit: def _shared_name() file: mesh_tensorflow/tpu_variables.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 991 unit: def _unique_id() file: mesh_tensorflow/tpu_variables.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 992 unit: def name() file: mesh_tensorflow/tpu_variables.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 993 unit: def dtype() file: mesh_tensorflow/tpu_variables.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 994 unit: def shape() file: mesh_tensorflow/tpu_variables.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 995 unit: def get_shape() file: mesh_tensorflow/tpu_variables.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 996 unit: def to_proto() file: mesh_tensorflow/tpu_variables.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 2 id: 997 unit: def constraint() file: mesh_tensorflow/tpu_variables.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 998 unit: def op() file: mesh_tensorflow/tpu_variables.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 999 unit: def read_value() file: mesh_tensorflow/tpu_variables.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 1000 unit: def get() file: mesh_tensorflow/tpu_variables.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 1001 unit: def _in_graph_mode() file: mesh_tensorflow/tpu_variables.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 1002 unit: def _should_act_as_resource_variable() file: mesh_tensorflow/tpu_variables.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 1003 unit: def _tensor_conversion() file: mesh_tensorflow/tpu_variables.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 4 id: 1004 unit: def to_json_string() file: mesh_tensorflow/bert/bert.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 1005 unit: def get_pooled_output() file: mesh_tensorflow/bert/bert.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 1006 unit: def get_sequence_output() file: mesh_tensorflow/bert/bert.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 1007 unit: def get_all_encoder_layers() file: mesh_tensorflow/bert/bert.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 1008 unit: def get_word_embedding_output() file: mesh_tensorflow/bert/bert.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 1009 unit: def get_embedding_output() file: mesh_tensorflow/bert/bert.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 1010 unit: def get_embedding_table() file: mesh_tensorflow/bert/bert.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 1011 unit: def get_extra_loss() file: mesh_tensorflow/bert/bert.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 1012 unit: def model_dim() file: mesh_tensorflow/bert/bert.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 1013 unit: def token_type_vocab_dim() file: mesh_tensorflow/bert/bert.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 1014 unit: def num_heads_dim() file: mesh_tensorflow/bert/bert.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 1015 unit: def softmax_heads_dims() file: mesh_tensorflow/bert/bert.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 1016 unit: def seq_dim() file: mesh_tensorflow/bert/bert.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 1017 unit: def memory_seq_dim() file: mesh_tensorflow/bert/bert.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 1018 unit: def __str__() file: mesh_tensorflow/bert/run_squad.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 1019 unit: def close() file: mesh_tensorflow/bert/run_squad.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 1020 unit: def get_train_examples() file: mesh_tensorflow/bert/run_classifier.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 2 id: 1021 unit: def get_dev_examples() file: mesh_tensorflow/bert/run_classifier.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 2 id: 1022 unit: def get_test_examples() file: mesh_tensorflow/bert/run_classifier.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 2 id: 1023 unit: def get_labels() file: mesh_tensorflow/bert/run_classifier.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 1024 unit: def __init__() file: mesh_tensorflow/bert/run_classifier.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 1025 unit: def get_labels() file: mesh_tensorflow/bert/run_classifier.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 1026 unit: def get_labels() file: mesh_tensorflow/bert/run_classifier.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 1027 unit: def get_labels() file: mesh_tensorflow/bert/run_classifier.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 1028 unit: def get_labels() file: mesh_tensorflow/bert/run_classifier.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 1029 unit: def convert_tokens_to_ids() file: mesh_tensorflow/bert/tokenization.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 2 id: 1030 unit: def convert_ids_to_tokens() file: mesh_tensorflow/bert/tokenization.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 2 id: 1031 unit: def convert_tokens_to_ids() file: mesh_tensorflow/bert/tokenization.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 2 id: 1032 unit: def convert_ids_to_tokens() file: mesh_tensorflow/bert/tokenization.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 2