id: 1
unit: def _get_bw_configs()
file: tzrec/ops/triton/triton_hstu_attention.py
start line: 0
end line: 0
size: 197 LOC
McCabe index: 9
number of parameters: 0

id: 2
unit: def _create_feature_config()
file: tzrec/tools/convert_easyrec_config_to_tzrec_config.py
start line: 0
end line: 0
size: 179 LOC
McCabe index: 34
number of parameters: 2

id: 3
unit: def _get_fw_configs()
file: tzrec/ops/triton/triton_hstu_attention.py
start line: 0
end line: 0
size: 170 LOC
McCabe index: 7
number of parameters: 0

id: 4
unit: def get_mm_configs()
file: tzrec/ops/triton/triton_addmm.py
start line: 0
end line: 0
size: 129 LOC
McCabe index: 14
number of parameters: 0

id: 5
unit: def edit_config()
file: tzrec/utils/config_util.py
start line: 0
end line: 0
size: 121 LOC
McCabe index: 37
number of parameters: 3

id: 6
unit: def _create_feature_config_no_fg()
file: tzrec/tools/convert_easyrec_config_to_tzrec_config.py
start line: 0
end line: 0
size: 99 LOC
McCabe index: 21
number of parameters: 2

id: 7
unit: def save()
file: tzrec/tools/tdm/gen_tree/tree_search_util.py
start line: 0
end line: 0
size: 94 LOC
McCabe index: 26
number of parameters: 3

id: 8
unit: def _check_graph_module()
file: tzrec/acc/export_utils.py
start line: 0
end line: 0
size: 88 LOC
McCabe index: 27
number of parameters: 2

id: 9
unit: def _convert_model_config()
file: tzrec/tools/convert_easyrec_config_to_tzrec_config.py
start line: 0
end line: 0
size: 86 LOC
McCabe index: 17
number of parameters: 3

id: 10
unit: def _build_batch()
file: tzrec/datasets/dataset.py
start line: 0
end line: 0
size: 86 LOC
McCabe index: 22
number of parameters: 3

id: 11
unit: def fg_json()
file: tzrec/features/lookup_feature.py
start line: 0
end line: 0
size: 74 LOC
McCabe index: 18
number of parameters: 1

id: 12
unit: def get()
file: tzrec/datasets/sampler.py
start line: 0
end line: 0
size: 63 LOC
McCabe index: 13
number of parameters: 3

id: 13
unit: def compute()
file: tzrec/metrics/grouped_auc.py
start line: 0
end line: 0
size: 60 LOC
McCabe index: 16
number of parameters: 1

id: 14
unit: def fg_json()
file: tzrec/features/match_feature.py
start line: 0
end line: 0
size: 54 LOC
McCabe index: 14
number of parameters: 1

id: 15
unit: def _train()
file: tzrec/tools/tdm/gen_tree/tree_cluster.py
start line: 0
end line: 0
size: 53 LOC
McCabe index: 15
number of parameters: 3

id: 16
unit: def _parse()
file: tzrec/features/lookup_feature.py
start line: 0
end line: 0
size: 48 LOC
McCabe index: 10
number of parameters: 3

id: 17
unit: def _parse()
file: tzrec/features/sequence_feature.py
start line: 0
end line: 0
size: 48 LOC
McCabe index: 7
number of parameters: 3

id: 18
unit: def _hstu_attn_bwd()
file: tzrec/ops/triton/triton_hstu_attention.py
start line: 0
end line: 0
size: 48 LOC
McCabe index: 1
number of parameters: 0

id: 19
unit: def _parse()
file: tzrec/features/sequence_feature.py
start line: 0
end line: 0
size: 45 LOC
McCabe index: 6
number of parameters: 3

id: 20
unit: def _feature_dim_dropout_ratio()
file: tzrec/tools/feature_selection.py
start line: 0
end line: 0
size: 44 LOC
McCabe index: 9
number of parameters: 1

id: 21
unit: def fg_json()
file: tzrec/features/tokenize_feature.py
start line: 0
end line: 0
size: 44 LOC
McCabe index: 8
number of parameters: 1

id: 22
unit: def parse()
file: tzrec/datasets/data_parser.py
start line: 0
end line: 0
size: 43 LOC
McCabe index: 13
number of parameters: 3

id: 23
unit: def _read()
file: tzrec/tools/tdm/gen_tree/tree_cluster.py
start line: 0
end line: 0
size: 42 LOC
McCabe index: 10
number of parameters: 1

id: 24
unit: def fg_json()
file: tzrec/features/custom_feature.py
start line: 0
end line: 0
size: 42 LOC
McCabe index: 11
number of parameters: 1

id: 25
unit: def forward()
file: tzrec/modules/sequence.py
start line: 0
end line: 0
size: 41 LOC
McCabe index: 3
number of parameters: 3

id: 26
unit: def fg_json()
file: tzrec/features/sequence_feature.py
start line: 0
end line: 0
size: 41 LOC
McCabe index: 12
number of parameters: 1

id: 27
unit: def fg_encoded_default_value()
file: tzrec/features/feature.py
start line: 0
end line: 0
size: 40 LOC
McCabe index: 11
number of parameters: 1

id: 28
unit: def _get_easyrec()
file: tzrec/tools/convert_easyrec_config_to_tzrec_config.py
start line: 0
end line: 0
size: 39 LOC
McCabe index: 6
number of parameters: 1

id: 29
unit: def _hstu_attn_fwd()
file: tzrec/ops/triton/triton_hstu_attention.py
start line: 0
end line: 0
size: 38 LOC
McCabe index: 1
number of parameters: 0

id: 30
unit: def _hstu_attn_fwd_persistent()
file: tzrec/ops/triton/triton_hstu_attention.py
start line: 0
end line: 0
size: 38 LOC
McCabe index: 1
number of parameters: 0

id: 31
unit: def _read()
file: tzrec/tools/tdm/gen_tree/tree_generator.py
start line: 0
end line: 0
size: 37 LOC
McCabe index: 8
number of parameters: 1

id: 32
unit: def _create_odps_account()
file: tzrec/datasets/odps_dataset.py
start line: 0
end line: 0
size: 37 LOC
McCabe index: 8
number of parameters: 0

id: 33
unit: def predict()
file: tzrec/models/dc2vr.py
start line: 0
end line: 0
size: 36 LOC
McCabe index: 9
number of parameters: 2

id: 34
unit: def _convert_model_feature_group()
file: tzrec/tools/convert_easyrec_config_to_tzrec_config.py
start line: 0
end line: 0
size: 36 LOC
McCabe index: 6
number of parameters: 2

id: 35
unit: def predict()
file: tzrec/models/dbmtl.py
start line: 0
end line: 0
size: 35 LOC
McCabe index: 9
number of parameters: 2

id: 36
unit: def _cluster()
file: tzrec/tools/tdm/gen_tree/tree_cluster.py
start line: 0
end line: 0
size: 35 LOC
McCabe index: 7
number of parameters: 2

id: 37
unit: def fg_json()
file: tzrec/features/id_feature.py
start line: 0
end line: 0
size: 35 LOC
McCabe index: 11
number of parameters: 1

id: 38
unit: def _init_session()
file: tzrec/datasets/odps_dataset.py
start line: 0
end line: 0
size: 34 LOC
McCabe index: 10
number of parameters: 1

id: 39
unit: def pin_memory()
file: tzrec/datasets/utils.py
start line: 0
end line: 0
size: 34 LOC
McCabe index: 10
number of parameters: 1

id: 40
unit: def _hstu_attn_bwd_one_col_block()
file: tzrec/ops/triton/triton_hstu_attention.py
start line: 0
end line: 0
size: 34 LOC
McCabe index: 1
number of parameters: 0

id: 41
unit: def mc_module()
file: tzrec/features/feature.py
start line: 0
end line: 0
size: 33 LOC
McCabe index: 8
number of parameters: 2

id: 42
unit: def _parse()
file: tzrec/features/match_feature.py
start line: 0
end line: 0
size: 33 LOC
McCabe index: 7
number of parameters: 3

id: 43
unit: def _hstu_attn_fwd_compute()
file: tzrec/ops/triton/triton_hstu_attention.py
start line: 0
end line: 0
size: 33 LOC
McCabe index: 1
number of parameters: 0

id: 44
unit: def save_predict_edge()
file: tzrec/tools/tdm/gen_tree/tree_search_util.py
start line: 0
end line: 0
size: 32 LOC
McCabe index: 12
number of parameters: 1

id: 45
unit: def _hstu_attn_bwd_one_block()
file: tzrec/ops/triton/triton_hstu_attention.py
start line: 0
end line: 0
size: 32 LOC
McCabe index: 1
number of parameters: 0

id: 46
unit: def forward()
file: tzrec/modules/sequence.py
start line: 0
end line: 0
size: 31 LOC
McCabe index: 2
number of parameters: 3

id: 47
unit: def auto_import()
file: tzrec/utils/load_class.py
start line: 0
end line: 0
size: 31 LOC
McCabe index: 12
number of parameters: 1

id: 48
unit: def create_planner()
file: tzrec/utils/plan_util.py
start line: 0
end line: 0
size: 31 LOC
McCabe index: 5
number of parameters: 2

id: 49
unit: def dump_parsed_inputs()
file: tzrec/datasets/data_parser.py
start line: 0
end line: 0
size: 31 LOC
McCabe index: 10
number of parameters: 3

id: 50
unit: def _bootstrap()
file: tzrec/datasets/sampler.py
start line: 0
end line: 0
size: 31 LOC
McCabe index: 7
number of parameters: 3

id: 51
unit: def _parse()
file: tzrec/features/custom_feature.py
start line: 0
end line: 0
size: 31 LOC
McCabe index: 8
number of parameters: 3

id: 52
unit: def gather_strings()
file: tzrec/utils/dist_util.py
start line: 0
end line: 0
size: 30 LOC
McCabe index: 10
number of parameters: 2

id: 53
unit: def _group_norm_mul_dropout_bwd_dx_du()
file: tzrec/ops/triton/triton_hstu_linear.py
start line: 0
end line: 0
size: 30 LOC
McCabe index: 1
number of parameters: 0

id: 54
unit: def to()
file: tzrec/datasets/utils.py
start line: 0
end line: 0
size: 29 LOC
McCabe index: 7
number of parameters: 3

id: 55
unit: def _parse()
file: tzrec/features/overlap_feature.py
start line: 0
end line: 0
size: 29 LOC
McCabe index: 7
number of parameters: 3

id: 56
unit: def _parse()
file: tzrec/features/id_feature.py
start line: 0
end line: 0
size: 29 LOC
McCabe index: 5
number of parameters: 3

id: 57
unit: def _add_timestamp_position_embeddings_kernel()
file: tzrec/ops/triton/triton_position.py
start line: 0
end line: 0
size: 29 LOC
McCabe index: 1
number of parameters: 0

id: 58
unit: def create_activation()
file: tzrec/modules/activation.py
start line: 0
end line: 0
size: 28 LOC
McCabe index: 8
number of parameters: 2

id: 59
unit: def _parse()
file: tzrec/features/sequence_feature.py
start line: 0
end line: 0
size: 28 LOC
McCabe index: 4
number of parameters: 3

id: 60
unit: def _ln_mul_dropout_bwd_dx_du()
file: tzrec/ops/triton/triton_hstu_linear.py
start line: 0
end line: 0
size: 28 LOC
McCabe index: 1
number of parameters: 0

id: 61
unit: def predict()
file: tzrec/models/dat.py
start line: 0
end line: 0
size: 27 LOC
McCabe index: 3
number of parameters: 2

id: 62
unit: def predict()
file: tzrec/models/rocket_launching.py
start line: 0
end line: 0
size: 27 LOC
McCabe index: 6
number of parameters: 2

id: 63
unit: def _parse_nodes()
file: tzrec/datasets/sampler.py
start line: 0
end line: 0
size: 27 LOC
McCabe index: 6
number of parameters: 2

id: 64
unit: def init()
file: tzrec/datasets/sampler.py
start line: 0
end line: 0
size: 27 LOC
McCabe index: 4
number of parameters: 2

id: 65
unit: def _parse()
file: tzrec/features/raw_feature.py
start line: 0
end line: 0
size: 27 LOC
McCabe index: 6
number of parameters: 3

id: 66
unit: def create_local_plan()
file: tzrec/utils/checkpoint_util.py
start line: 0
end line: 0
size: 26 LOC
McCabe index: 8
number of parameters: 1

id: 67
unit: def _write_loop()
file: tzrec/tools/tdm/retrieval.py
start line: 0
end line: 0
size: 26 LOC
McCabe index: 6
number of parameters: 2

id: 68
unit: def _init_writer()
file: tzrec/datasets/odps_dataset.py
start line: 0
end line: 0
size: 26 LOC
McCabe index: 5
number of parameters: 1

id: 69
unit: def fg_json()
file: tzrec/features/combo_feature.py
start line: 0
end line: 0
size: 26 LOC
McCabe index: 7
number of parameters: 1

id: 70
unit: def run_cmd()
file: tzrec/utils/misc_util.py
start line: 0
end line: 0
size: 25 LOC
McCabe index: 6
number of parameters: 4

id: 71
unit: def predict()
file: tzrec/models/hstu.py
start line: 0
end line: 0
size: 25 LOC
McCabe index: 2
number of parameters: 2

id: 72
unit: def train()
file: tzrec/tools/tdm/gen_tree/tree_cluster.py
start line: 0
end line: 0
size: 24 LOC
McCabe index: 7
number of parameters: 2

id: 73
unit: def _easyrec_metrics_2_tzrec_metrics()
file: tzrec/tools/convert_easyrec_config_to_tzrec_config.py
start line: 0
end line: 0
size: 24 LOC
McCabe index: 7
number of parameters: 2

id: 74
unit: def _parse()
file: tzrec/features/expr_feature.py
start line: 0
end line: 0
size: 24 LOC
McCabe index: 6
number of parameters: 3

id: 75
unit: def _parse()
file: tzrec/features/tokenize_feature.py
start line: 0
end line: 0
size: 24 LOC
McCabe index: 5
number of parameters: 3

id: 76
unit: def _get_basic_types()
file: tzrec/utils/config_util.py
start line: 0
end line: 0
size: 22 LOC
McCabe index: 1
number of parameters: 0

id: 77
unit: def init_input()
file: tzrec/models/match_model.py
start line: 0
end line: 0
size: 22 LOC
McCabe index: 5
number of parameters: 1

id: 78
unit: def process()
file: tzrec/tools/feature_selection.py
start line: 0
end line: 0
size: 22 LOC
McCabe index: 5
number of parameters: 1

id: 79
unit: def fg_json()
file: tzrec/features/sequence_feature.py
start line: 0
end line: 0
size: 22 LOC
McCabe index: 7
number of parameters: 1

id: 80
unit: def hstu_preprocess_and_attention()
file: tzrec/ops/hstu_compute.py
start line: 0
end line: 0
size: 22 LOC
McCabe index: 1
number of parameters: 0

id: 81
unit: def _addmm_fwd()
file: tzrec/ops/triton/triton_addmm.py
start line: 0
end line: 0
size: 22 LOC
McCabe index: 1
number of parameters: 0

id: 82
unit: def forward()
file: tzrec/modules/sequence.py
start line: 0
end line: 0
size: 21 LOC
McCabe index: 2
number of parameters: 3

id: 83
unit: def list_distcp_param()
file: tzrec/utils/checkpoint_util.py
start line: 0
end line: 0
size: 21 LOC
McCabe index: 8
number of parameters: 1

id: 84
unit: def import_pkg()
file: tzrec/utils/load_class.py
start line: 0
end line: 0
size: 21 LOC
McCabe index: 7
number of parameters: 2

id: 85
unit: def predict()
file: tzrec/models/deepfm.py
start line: 0
end line: 0
size: 21 LOC
McCabe index: 3
number of parameters: 2

id: 86
unit: def init_input()
file: tzrec/models/rank_model.py
start line: 0
end line: 0
size: 21 LOC
McCabe index: 5
number of parameters: 1

id: 87
unit: def _easyrec_bayes_tower_2_tzrec_bayes_tower()
file: tzrec/tools/convert_easyrec_config_to_tzrec_config.py
start line: 0
end line: 0
size: 21 LOC
McCabe index: 3
number of parameters: 2

id: 88
unit: def _load_feature_info()
file: tzrec/tools/add_feature_info_to_config.py
start line: 0
end line: 0
size: 21 LOC
McCabe index: 5
number of parameters: 1

id: 89
unit: def _jagged_jagged_bmm_reduce_sum()
file: tzrec/ops/triton/triton_jagged_tensors.py
start line: 0
end line: 0
size: 21 LOC
McCabe index: 1
number of parameters: 0

id: 90
unit: def forward()
file: tzrec/ops/triton/triton_hstu_preprocess_and_attention.py
start line: 0
end line: 0
size: 21 LOC
McCabe index: 1
number of parameters: 0

id: 91
unit: def _init_tree()
file: tzrec/tools/tdm/gen_tree/tree_generator.py
start line: 0
end line: 0
size: 20 LOC
McCabe index: 4
number of parameters: 3

id: 92
unit: def num_embeddings()
file: tzrec/features/id_feature.py
start line: 0
end line: 0
size: 20 LOC
McCabe index: 7
number of parameters: 1

id: 93
unit: def _parse()
file: tzrec/features/combo_feature.py
start line: 0
end line: 0
size: 20 LOC
McCabe index: 5
number of parameters: 3

id: 94
unit: def _group_norm_mul_dropout_fwd()
file: tzrec/ops/triton/triton_hstu_linear.py
start line: 0
end line: 0
size: 20 LOC
McCabe index: 1
number of parameters: 0

id: 95
unit: def jagged_dense_bmm_broadcast_add_kernel()
file: tzrec/ops/triton/triton_jagged_tensors.py
start line: 0
end line: 0
size: 20 LOC
McCabe index: 1
number of parameters: 0

id: 96
unit: def triton_hstu_preprocess_and_attention()
file: tzrec/ops/triton/triton_hstu_preprocess_and_attention.py
start line: 0
end line: 0
size: 20 LOC
McCabe index: 1
number of parameters: 0

id: 97
unit: def sample_noisy_input()
file: tzrec/modules/variational_dropout.py
start line: 0
end line: 0
size: 19 LOC
McCabe index: 2
number of parameters: 2

id: 98
unit: def get_lr()
file: tzrec/optim/lr_scheduler.py
start line: 0
end line: 0
size: 19 LOC
McCabe index: 5
number of parameters: 1

id: 99
unit: def _parse_odps_config_file()
file: tzrec/datasets/odps_dataset.py
start line: 0
end line: 0
size: 19 LOC
McCabe index: 5
number of parameters: 1

id: 100
unit: def vocab_dict()
file: tzrec/features/feature.py
start line: 0
end line: 0
size: 19 LOC
McCabe index: 7
number of parameters: 1

id: 101
unit: def main()
file: tzrec/ops/benchmarks/hstu_attention_bench.py
start line: 0
end line: 0
size: 19 LOC
McCabe index: 1
number of parameters: 0

id: 102
unit: def triton_group_norm_mul_dropout_bwd()
file: tzrec/ops/triton/triton_hstu_linear.py
start line: 0
end line: 0
size: 19 LOC
McCabe index: 1
number of parameters: 0

id: 103
unit: def _get_bmm_configs()
file: tzrec/ops/triton/triton_jagged_tensors.py
start line: 0
end line: 0
size: 19 LOC
McCabe index: 6
number of parameters: 0

id: 104
unit: def _hstu_attn_fwd_one_block()
file: tzrec/ops/triton/triton_hstu_attention.py
start line: 0
end line: 0
size: 19 LOC
McCabe index: 1
number of parameters: 0

id: 105
unit: def __init__()
file: tzrec/modules/sequence.py
start line: 0
end line: 0
size: 18 LOC
McCabe index: 1
number of parameters: 0

id: 106
unit: def forward()
file: tzrec/modules/dense_embedding_collection.py
start line: 0
end line: 0
size: 18 LOC
McCabe index: 3
number of parameters: 2

id: 107
unit: def _column_modes()
file: tzrec/tools/tdm/gen_tree/tree_builder.py
start line: 0
end line: 0
size: 18 LOC
McCabe index: 7
number of parameters: 2

id: 108
unit: def vocab_list()
file: tzrec/features/feature.py
start line: 0
end line: 0
size: 18 LOC
McCabe index: 5
number of parameters: 1

id: 109
unit: def num_embeddings()
file: tzrec/features/combo_feature.py
start line: 0
end line: 0
size: 18 LOC
McCabe index: 6
number of parameters: 1

id: 110
unit: def quant_dtype()
file: tzrec/acc/utils.py
start line: 0
end line: 0
size: 18 LOC
McCabe index: 4
number of parameters: 0

id: 111
unit: def _bench_hstu_attention()
file: tzrec/ops/benchmarks/hstu_attention_bench.py
start line: 0
end line: 0
size: 18 LOC
McCabe index: 1
number of parameters: 0

id: 112
unit: def _ln_mul_dropout_fwd()
file: tzrec/ops/triton/triton_hstu_linear.py
start line: 0
end line: 0
size: 18 LOC
McCabe index: 1
number of parameters: 0

id: 113
unit: def _weighted_layer_norm_bwd_dx()
file: tzrec/ops/triton/triton_layer_norm.py
start line: 0
end line: 0
size: 18 LOC
McCabe index: 1
number of parameters: 0

id: 114
unit: def parse_requirements()
file: setup.py
start line: 0
end line: 0
size: 17 LOC
McCabe index: 7
number of parameters: 1

id: 115
unit: def get_register_class_meta()
file: tzrec/utils/load_class.py
start line: 0
end line: 0
size: 17 LOC
McCabe index: 2
number of parameters: 1

id: 116
unit: def init_parameters()
file: tzrec/utils/state_dict_util.py
start line: 0
end line: 0
size: 17 LOC
McCabe index: 12
number of parameters: 2

id: 117
unit: def build_input()
file: tzrec/models/rank_model.py
start line: 0
end line: 0
size: 17 LOC
McCabe index: 3
number of parameters: 2

id: 118
unit: def build_input()
file: tzrec/models/match_model.py
start line: 0
end line: 0
size: 17 LOC
McCabe index: 3
number of parameters: 2

id: 119
unit: def _forward_loop()
file: tzrec/tools/tdm/retrieval.py
start line: 0
end line: 0
size: 17 LOC
McCabe index: 5
number of parameters: 3

id: 120
unit: def _get_nodes()
file: tzrec/tools/tdm/gen_tree/tree_search_util.py
start line: 0
end line: 0
size: 17 LOC
McCabe index: 5
number of parameters: 1

id: 121
unit: def close()
file: tzrec/datasets/odps_dataset.py
start line: 0
end line: 0
size: 17 LOC
McCabe index: 6
number of parameters: 1

id: 122
unit: def num_embeddings()
file: tzrec/features/lookup_feature.py
start line: 0
end line: 0
size: 17 LOC
McCabe index: 7
number of parameters: 1

id: 123
unit: def num_embeddings()
file: tzrec/features/custom_feature.py
start line: 0
end line: 0
size: 17 LOC
McCabe index: 7
number of parameters: 1

id: 124
unit: def fg_json()
file: tzrec/features/raw_feature.py
start line: 0
end line: 0
size: 17 LOC
McCabe index: 5
number of parameters: 1

id: 125
unit: def num_embeddings()
file: tzrec/features/match_feature.py
start line: 0
end line: 0
size: 17 LOC
McCabe index: 7
number of parameters: 1

id: 126
unit: def forward()
file: tzrec/ops/triton/triton_hstu_linear.py
start line: 0
end line: 0
size: 17 LOC
McCabe index: 1
number of parameters: 0

id: 127
unit: def forward()
file: tzrec/modules/mmoe.py
start line: 0
end line: 0
size: 16 LOC
McCabe index: 4
number of parameters: 2

id: 128
unit: def latest_checkpoint()
file: tzrec/utils/checkpoint_util.py
start line: 0
end line: 0
size: 16 LOC
McCabe index: 6
number of parameters: 1

id: 129
unit: def propose()
file: tzrec/utils/plan_util.py
start line: 0
end line: 0
size: 16 LOC
McCabe index: 5
number of parameters: 1

id: 130
unit: def broadcast_string()
file: tzrec/utils/dist_util.py
start line: 0
end line: 0
size: 16 LOC
McCabe index: 5
number of parameters: 2

id: 131
unit: def _create_mean_pooling_divisor()
file: tzrec/utils/dist_util.py
start line: 0
end line: 0
size: 16 LOC
McCabe index: 1
number of parameters: 0

id: 132
unit: def _easyrec_loss_2_tzrec_loss()
file: tzrec/tools/convert_easyrec_config_to_tzrec_config.py
start line: 0
end line: 0
size: 16 LOC
McCabe index: 5
number of parameters: 2

id: 133
unit: def emb_bag_config()
file: tzrec/features/feature.py
start line: 0
end line: 0
size: 16 LOC
McCabe index: 4
number of parameters: 1

id: 134
unit: def side_inputs()
file: tzrec/features/feature.py
start line: 0
end line: 0
size: 16 LOC
McCabe index: 6
number of parameters: 1

id: 135
unit: def init_fg()
file: tzrec/features/sequence_feature.py
start line: 0
end line: 0
size: 16 LOC
McCabe index: 2
number of parameters: 1

id: 136
unit: def init_fg()
file: tzrec/features/sequence_feature.py
start line: 0
end line: 0
size: 16 LOC
McCabe index: 2
number of parameters: 1

id: 137
unit: def init_fg()
file: tzrec/features/sequence_feature.py
start line: 0
end line: 0
size: 16 LOC
McCabe index: 2
number of parameters: 1

id: 138
unit: def hstu_compute_output()
file: tzrec/ops/hstu_compute.py
start line: 0
end line: 0
size: 16 LOC
McCabe index: 1
number of parameters: 0

id: 139
unit: def hstu_mha()
file: tzrec/ops/hstu_attention.py
start line: 0
end line: 0
size: 16 LOC
McCabe index: 1
number of parameters: 0

id: 140
unit: def triton_layer_norm_mul_dropout_bwd()
file: tzrec/ops/triton/triton_hstu_linear.py
start line: 0
end line: 0
size: 16 LOC
McCabe index: 1
number of parameters: 0

id: 141
unit: def triton_hstu_compute_output()
file: tzrec/ops/triton/triton_hstu_linear.py
start line: 0
end line: 0
size: 16 LOC
McCabe index: 1
number of parameters: 0

id: 142
unit: def _concat_2D_jagged()
file: tzrec/ops/triton/triton_jagged_tensors.py
start line: 0
end line: 0
size: 16 LOC
McCabe index: 1
number of parameters: 0

id: 143
unit: def _split_2D_jagged()
file: tzrec/ops/triton/triton_jagged_tensors.py
start line: 0
end line: 0
size: 16 LOC
McCabe index: 1
number of parameters: 0

id: 144
unit: def _add_embeddings_bwd_configs()
file: tzrec/ops/triton/triton_position.py
start line: 0
end line: 0
size: 16 LOC
McCabe index: 4
number of parameters: 0

id: 145
unit: def triton_hstu_attention_bwd()
file: tzrec/ops/triton/triton_hstu_attention.py
start line: 0
end line: 0
size: 16 LOC
McCabe index: 1
number of parameters: 0

id: 146
unit: def __init__()
file: tzrec/modules/hstu.py
start line: 0
end line: 0
size: 15 LOC
McCabe index: 1
number of parameters: 0

id: 147
unit: def load_by_path()
file: tzrec/utils/load_class.py
start line: 0
end line: 0
size: 15 LOC
McCabe index: 6
number of parameters: 1

id: 148
unit: def fix_mch_state()
file: tzrec/utils/state_dict_util.py
start line: 0
end line: 0
size: 15 LOC
McCabe index: 4
number of parameters: 1

id: 149
unit: def predict()
file: tzrec/models/ple.py
start line: 0
end line: 0
size: 15 LOC
McCabe index: 3
number of parameters: 2

id: 150
unit: def predict()
file: tzrec/models/dlrm.py
start line: 0
end line: 0
size: 15 LOC
McCabe index: 2
number of parameters: 2

id: 151
unit: def get()
file: tzrec/datasets/sampler.py
start line: 0
end line: 0
size: 15 LOC
McCabe index: 2
number of parameters: 3

id: 152
unit: def _create_table()
file: tzrec/datasets/odps_dataset.py
start line: 0
end line: 0
size: 15 LOC
McCabe index: 5
number of parameters: 3

id: 153
unit: def get_worker_info()
file: tzrec/datasets/dataset.py
start line: 0
end line: 0
size: 15 LOC
McCabe index: 3
number of parameters: 1

id: 154
unit: def fg_json()
file: tzrec/features/overlap_feature.py
start line: 0
end line: 0
size: 15 LOC
McCabe index: 3
number of parameters: 1

id: 155
unit: def emb_config()
file: tzrec/features/feature.py
start line: 0
end line: 0
size: 15 LOC
McCabe index: 4
number of parameters: 1

id: 156
unit: def operator_lib_file()
file: tzrec/features/custom_feature.py
start line: 0
end line: 0
size: 15 LOC
McCabe index: 4
number of parameters: 1

id: 157
unit: def _add_position_embeddings_configs()
file: tzrec/ops/triton/triton_position.py
start line: 0
end line: 0
size: 15 LOC
McCabe index: 4
number of parameters: 0

id: 158
unit: def _add_position_embeddings_kernel()
file: tzrec/ops/triton/triton_position.py
start line: 0
end line: 0
size: 15 LOC
McCabe index: 1
number of parameters: 0

id: 159
unit: def _weighted_layer_norm_fwd()
file: tzrec/ops/triton/triton_layer_norm.py
start line: 0
end line: 0
size: 15 LOC
McCabe index: 1
number of parameters: 0

id: 160
unit: def _weighted_rms_norm_bwd_dx()
file: tzrec/ops/triton/triton_layer_norm.py
start line: 0
end line: 0
size: 15 LOC
McCabe index: 1
number of parameters: 0

id: 161
unit: def forward()
file: tzrec/modules/masknet_module.py
start line: 0
end line: 0
size: 14 LOC
McCabe index: 4
number of parameters: 2

id: 162
unit: def _hstu_attention_maybe_from_cache()
file: tzrec/modules/hstu.py
start line: 0
end line: 0
size: 14 LOC
McCabe index: 1
number of parameters: 0

id: 163
unit: def get_lr()
file: tzrec/optim/lr_scheduler.py
start line: 0
end line: 0
size: 14 LOC
McCabe index: 5
number of parameters: 1

id: 164
unit: def _get_compatible_fg_mode()
file: tzrec/utils/config_util.py
start line: 0
end line: 0
size: 14 LOC
McCabe index: 4
number of parameters: 1

id: 165
unit: def sparse_parameters()
file: tzrec/models/model.py
start line: 0
end line: 0
size: 14 LOC
McCabe index: 5
number of parameters: 1

id: 166
unit: def _easyrec_task_tower_2_tzrec_task_tower()
file: tzrec/tools/convert_easyrec_config_to_tzrec_config.py
start line: 0
end line: 0
size: 14 LOC
McCabe index: 3
number of parameters: 2

id: 167
unit: def init()
file: tzrec/datasets/sampler.py
start line: 0
end line: 0
size: 14 LOC
McCabe index: 3
number of parameters: 2

id: 168
unit: def get()
file: tzrec/datasets/sampler.py
start line: 0
end line: 0
size: 14 LOC
McCabe index: 2
number of parameters: 3

id: 169
unit: def _init_client()
file: tzrec/datasets/odps_dataset.py
start line: 0
end line: 0
size: 14 LOC
McCabe index: 4
number of parameters: 1

id: 170
unit: def write()
file: tzrec/datasets/odps_dataset.py
start line: 0
end line: 0
size: 14 LOC
McCabe index: 3
number of parameters: 3

id: 171
unit: def _init_input_fields()
file: tzrec/datasets/odps_dataset_v1.py
start line: 0
end line: 0
size: 14 LOC
McCabe index: 2
number of parameters: 1

id: 172
unit: def write()
file: tzrec/datasets/parquet_dataset.py
start line: 0
end line: 0
size: 14 LOC
McCabe index: 3
number of parameters: 3

id: 173
unit: def fg_json()
file: tzrec/features/sequence_feature.py
start line: 0
end line: 0
size: 14 LOC
McCabe index: 3
number of parameters: 1

id: 174
unit: def _softmax()
file: tzrec/acc/export_utils.py
start line: 0
end line: 0
size: 14 LOC
McCabe index: 3
number of parameters: 3

id: 175
unit: def pytorch_hstu_compute_output()
file: tzrec/ops/pytorch/pt_hstu_linear.py
start line: 0
end line: 0
size: 14 LOC
McCabe index: 1
number of parameters: 0

id: 176
unit: def pytorch_hstu_mha()
file: tzrec/ops/pytorch/pt_hstu_attention.py
start line: 0
end line: 0
size: 14 LOC
McCabe index: 1
number of parameters: 0

id: 177
unit: def add_timestamp_positional_embeddings()
file: tzrec/ops/position.py
start line: 0
end line: 0
size: 14 LOC
McCabe index: 1
number of parameters: 0

id: 178
unit: def _add_position_embeddings_bwd_kernel()
file: tzrec/ops/triton/triton_position.py
start line: 0
end line: 0
size: 14 LOC
McCabe index: 1
number of parameters: 0

id: 179
unit: def _add_embeddings_bwd_kernel()
file: tzrec/ops/triton/triton_position.py
start line: 0
end line: 0
size: 14 LOC
McCabe index: 1
number of parameters: 0

id: 180
unit: def forward()
file: tzrec/modules/sequence.py
start line: 0
end line: 0
size: 13 LOC
McCabe index: 1
number of parameters: 3

id: 181
unit: def forward()
file: tzrec/modules/dense_embedding_collection.py
start line: 0
end line: 0
size: 13 LOC
McCabe index: 1
number of parameters: 2

id: 182
unit: def _dot_split_with_bracket()
file: tzrec/utils/config_util.py
start line: 0
end line: 0
size: 13 LOC
McCabe index: 3
number of parameters: 1

id: 183
unit: def predict()
file: tzrec/models/mind.py
start line: 0
end line: 0
size: 13 LOC
McCabe index: 3
number of parameters: 2

id: 184
unit: def predict()
file: tzrec/models/dssm.py
start line: 0
end line: 0
size: 13 LOC
McCabe index: 1
number of parameters: 2

id: 185
unit: def __init__()
file: tzrec/models/match_model.py
start line: 0
end line: 0
size: 13 LOC
McCabe index: 1
number of parameters: 0

id: 186
unit: def tdm_retrieval()
file: tzrec/tools/tdm/retrieval.py
start line: 0
end line: 0
size: 13 LOC
McCabe index: 1
number of parameters: 0

id: 187
unit: def build()
file: tzrec/tools/convert_easyrec_config_to_tzrec_config.py
start line: 0
end line: 0
size: 13 LOC
McCabe index: 2
number of parameters: 1

id: 188
unit: def init()
file: tzrec/datasets/sampler.py
start line: 0
end line: 0
size: 13 LOC
McCabe index: 2
number of parameters: 2

id: 189
unit: def _parse_table_path()
file: tzrec/datasets/odps_dataset.py
start line: 0
end line: 0
size: 13 LOC
McCabe index: 4
number of parameters: 1

id: 190
unit: def __init__()
file: tzrec/datasets/odps_dataset.py
start line: 0
end line: 0
size: 13 LOC
McCabe index: 1
number of parameters: 0

id: 191
unit: def record_stream()
file: tzrec/datasets/utils.py
start line: 0
end line: 0
size: 13 LOC
McCabe index: 7
number of parameters: 2

id: 192
unit: def _build_side_inputs()
file: tzrec/features/match_feature.py
start line: 0
end line: 0
size: 13 LOC
McCabe index: 6
number of parameters: 1

id: 193
unit: def _build_side_inputs()
file: tzrec/features/sequence_feature.py
start line: 0
end line: 0
size: 13 LOC
McCabe index: 5
number of parameters: 1

id: 194
unit: def predict()
file: tzrec/main.py
start line: 0
end line: 0
size: 13 LOC
McCabe index: 1
number of parameters: 0

id: 195
unit: def forward()
file: tzrec/ops/triton/triton_hstu_linear.py
start line: 0
end line: 0
size: 13 LOC
McCabe index: 1
number of parameters: 0

id: 196
unit: def triton_norm_mul_dropout()
file: tzrec/ops/triton/triton_hstu_linear.py
start line: 0
end line: 0
size: 13 LOC
McCabe index: 1
number of parameters: 0

id: 197
unit: def forward()
file: tzrec/ops/triton/triton_position.py
start line: 0
end line: 0
size: 13 LOC
McCabe index: 1
number of parameters: 0

id: 198
unit: def forward()
file: tzrec/ops/triton/triton_hstu_attention.py
start line: 0
end line: 0
size: 13 LOC
McCabe index: 1
number of parameters: 0

id: 199
unit: def group_dims()
file: tzrec/modules/embedding.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 5
number of parameters: 2

id: 200
unit: def predict()
file: tzrec/models/multi_tower_din_trt.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 4
number of parameters: 2

id: 201
unit: def predict()
file: tzrec/models/multi_tower_din.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 4
number of parameters: 2

id: 202
unit: def sim()
file: tzrec/models/match_model.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 2
number of parameters: 3

id: 203
unit: def analyse_fg()
file: tzrec/tools/convert_easyrec_config_to_tzrec_config.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 4
number of parameters: 2

id: 204
unit: def _wait_init_table()
file: tzrec/datasets/odps_dataset.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 5
number of parameters: 1

id: 205
unit: def __init__()
file: tzrec/datasets/csv_dataset.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 1
number of parameters: 0

id: 206
unit: def write()
file: tzrec/datasets/csv_dataset.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 3
number of parameters: 3

id: 207
unit: def fg_json()
file: tzrec/features/expr_feature.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 2
number of parameters: 1

id: 208
unit: def is_sparse()
file: tzrec/features/lookup_feature.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 8
number of parameters: 1

id: 209
unit: def vocab_file()
file: tzrec/features/feature.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 4
number of parameters: 1

id: 210
unit: def is_sparse()
file: tzrec/features/custom_feature.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 8
number of parameters: 1

id: 211
unit: def init_fg()
file: tzrec/features/tokenize_feature.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 4
number of parameters: 1

id: 212
unit: def is_sparse()
file: tzrec/features/match_feature.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 8
number of parameters: 1

id: 213
unit: def _train_and_evaluate()
file: tzrec/main.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 1
number of parameters: 0

id: 214
unit: def _get_config_paths()
file: tzrec/benchmark/benchmark.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 6
number of parameters: 1

id: 215
unit: def pytorch_norm_mul_dropout()
file: tzrec/ops/pytorch/pt_hstu_linear.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 1
number of parameters: 0

id: 216
unit: def pytorch_add_timestamp_positional_embeddings()
file: tzrec/ops/pytorch/pt_position.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 1
number of parameters: 0

id: 217
unit: def triton_group_norm_mul_dropout_fwd()
file: tzrec/ops/triton/triton_hstu_linear.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 1
number of parameters: 0

id: 218
unit: def triton_add_timestamp_positional_embeddings()
file: tzrec/ops/triton/triton_position.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 1
number of parameters: 0

id: 219
unit: def triton_hstu_attention_fwd()
file: tzrec/ops/triton/triton_hstu_attention.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 1
number of parameters: 0

id: 220
unit: def triton_hstu_mha()
file: tzrec/ops/triton/triton_hstu_attention.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 1
number of parameters: 0

id: 221
unit: def _layer_norm_fwd()
file: tzrec/ops/triton/triton_layer_norm.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 1
number of parameters: 0

id: 222
unit: def _layer_norm_bwd_dx()
file: tzrec/ops/triton/triton_layer_norm.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 1
number of parameters: 0

id: 223
unit: def __init__()
file: tzrec/modules/interaction.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 1
number of parameters: 3

id: 224
unit: def squash()
file: tzrec/modules/capsule.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 1
number of parameters: 2

id: 225
unit: def _get_checkpoint_step()
file: tzrec/utils/checkpoint_util.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 3
number of parameters: 1

id: 226
unit: def __init__()
file: tzrec/utils/plan_util.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 2
number of parameters: 2

id: 227
unit: def build_faiss_index()
file: tzrec/utils/faiss_util.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 1
number of parameters: 0

id: 228
unit: def predict()
file: tzrec/models/dssm_v2.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 1
number of parameters: 2

id: 229
unit: def init_loss()
file: tzrec/models/multi_task_rank.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 4
number of parameters: 1

id: 230
unit: def init_metric()
file: tzrec/models/multi_task_rank.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 4
number of parameters: 1

id: 231
unit: def __init__()
file: tzrec/tools/tdm/gen_tree/tree_cluster.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 1
number of parameters: 0

id: 232
unit: def _create_model_config()
file: tzrec/tools/convert_easyrec_config_to_tzrec_config.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 1
number of parameters: 2

id: 233
unit: def _get_gl_type()
file: tzrec/datasets/sampler.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 2
number of parameters: 1

id: 234
unit: def _get_np_type()
file: tzrec/datasets/sampler.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 2
number of parameters: 1

id: 235
unit: def _type_pa_to_table()
file: tzrec/datasets/odps_dataset.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 5
number of parameters: 1

id: 236
unit: def __init__()
file: tzrec/datasets/parquet_dataset.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 4
number of parameters: 3

id: 237
unit: def _init_input_fields()
file: tzrec/datasets/dataset.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 3
number of parameters: 1

id: 238
unit: def __init__()
file: tzrec/datasets/csv_dataset.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 4
number of parameters: 3

id: 239
unit: def init_process_group()
file: tzrec/main.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 2
number of parameters: 0

id: 240
unit: def export_acc_config()
file: tzrec/acc/utils.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 5
number of parameters: 0

id: 241
unit: def hstu_compute_uqvk()
file: tzrec/ops/hstu_compute.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 1
number of parameters: 0

id: 242
unit: def delta_hstu_mha()
file: tzrec/ops/hstu_attention.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 1
number of parameters: 0

id: 243
unit: def autotune_max_seq_len()
file: tzrec/ops/utils.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 5
number of parameters: 1

id: 244
unit: def _get_bwd_dwdb_configs()
file: tzrec/ops/triton/triton_hstu_linear.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 4
number of parameters: 0

id: 245
unit: def forward()
file: tzrec/ops/triton/triton_hstu_linear.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 1
number of parameters: 0

id: 246
unit: def _get_bwd_dwdb_configs()
file: tzrec/ops/triton/triton_hstu_linear.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 4
number of parameters: 0

id: 247
unit: def _get_bwd_dwdb_configs()
file: tzrec/ops/triton/triton_layer_norm.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 4
number of parameters: 0

id: 248
unit: def triton_weighted_layer_norm_bwd()
file: tzrec/ops/triton/triton_layer_norm.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 1
number of parameters: 0

id: 249
unit: def forward()
file: tzrec/modules/sequence.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 2
number of parameters: 3

id: 250
unit: def __init__()
file: tzrec/modules/extraction_net.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 1
number of parameters: 0

id: 251
unit: def forward()
file: tzrec/modules/activation.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 2
number of parameters: 2

id: 252
unit: def __init__()
file: tzrec/modules/mlp.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 1
number of parameters: 0

id: 253
unit: def forward()
file: tzrec/modules/variational_dropout.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 1
number of parameters: 2

id: 254
unit: def __init__()
file: tzrec/optim/lr_scheduler.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 1
number of parameters: 0

id: 255
unit: def __init__()
file: tzrec/models/mind.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 1
number of parameters: 0

id: 256
unit: def _get_distillation_mlp_index()
file: tzrec/models/rocket_launching.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 4
number of parameters: 1

id: 257
unit: def init_loss()
file: tzrec/models/rocket_launching.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 3
number of parameters: 1

id: 258
unit: def predict()
file: tzrec/models/multi_tower.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 3
number of parameters: 2

id: 259
unit: def __init__()
file: tzrec/models/match_model.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 1
number of parameters: 0

id: 260
unit: def _init_metric_impl()
file: tzrec/models/match_model.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 2
number of parameters: 3

id: 261
unit: def __init__()
file: tzrec/tools/tdm/gen_tree/tree_generator.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 1
number of parameters: 0

id: 262
unit: def __init__()
file: tzrec/datasets/parquet_dataset.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 1
number of parameters: 0

id: 263
unit: def is_user_feat()
file: tzrec/features/feature.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 5
number of parameters: 1

id: 264
unit: def num_embeddings()
file: tzrec/features/tokenize_feature.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 3
number of parameters: 1

id: 265
unit: def _forward()
file: tzrec/main.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 3
number of parameters: 1

id: 266
unit: def pytorch_cached_hstu_mha()
file: tzrec/ops/pytorch/pt_hstu_attention.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 1
number of parameters: 0

id: 267
unit: def add_positional_embeddings()
file: tzrec/ops/position.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 1
number of parameters: 0

id: 268
unit: def triton_layer_norm_mul_dropout_fwd()
file: tzrec/ops/triton/triton_hstu_linear.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 1
number of parameters: 0

id: 269
unit: def triton_cached_hstu_mha()
file: tzrec/ops/triton/triton_hstu_attention.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 1
number of parameters: 0

id: 270
unit: def _weighted_rms_norm_fwd()
file: tzrec/ops/triton/triton_layer_norm.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 1
number of parameters: 0

id: 271
unit: def __init__()
file: tzrec/metrics/recall_at_k.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 1
number of parameters: 3

id: 272
unit: def jagged_forward()
file: tzrec/modules/sequence.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 1
number of parameters: 0

id: 273
unit: def _dense_to_jt()
file: tzrec/modules/embedding.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 1
number of parameters: 1

id: 274
unit: def forward()
file: tzrec/modules/embedding.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 1
number of parameters: 0

id: 275
unit: def __init__()
file: tzrec/modules/mlp.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 1
number of parameters: 0

id: 276
unit: def forward()
file: tzrec/modules/hstu.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 1
number of parameters: 0

id: 277
unit: def get_default_sharders()
file: tzrec/utils/plan_util.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 4
number of parameters: 0

id: 278
unit: def log()
file: tzrec/utils/logging_util.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 3
number of parameters: 3

id: 279
unit: def DistributedModelParallel()
file: tzrec/utils/dist_util.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 1
number of parameters: 0

id: 280
unit: def _loss_impl()
file: tzrec/models/rank_model.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 1
number of parameters: 0

id: 281
unit: def predict()
file: tzrec/models/tdm.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 2
number of parameters: 2

id: 282
unit: def predict()
file: tzrec/models/mmoe.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 2
number of parameters: 2

id: 283
unit: def __init__()
file: tzrec/models/hstu.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 1
number of parameters: 0

id: 284
unit: def _tdm_predict_data_worker()
file: tzrec/tools/tdm/retrieval.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 1
number of parameters: 0

id: 285
unit: def __init__()
file: tzrec/tools/tdm/gen_tree/tree_builder.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 1
number of parameters: 0

id: 286
unit: def _column_means()
file: tzrec/tools/tdm/gen_tree/tree_builder.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 3
number of parameters: 2

id: 287
unit: def _create_data_config()
file: tzrec/tools/convert_easyrec_config_to_tzrec_config.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 1
number of parameters: 2

id: 288
unit: def __init__()
file: tzrec/tools/feature_selection.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 1
number of parameters: 0

id: 289
unit: def build()
file: tzrec/tools/add_feature_info_to_config.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 1
number of parameters: 1

id: 290
unit: def get()
file: tzrec/datasets/sampler.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 1
number of parameters: 3

id: 291
unit: def init()
file: tzrec/datasets/sampler.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 1
number of parameters: 2

id: 292
unit: def init()
file: tzrec/datasets/sampler.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 1
number of parameters: 2

id: 293
unit: def _refresh_sessions_daemon()
file: tzrec/datasets/odps_dataset.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 4
number of parameters: 2

id: 294
unit: def __init__()
file: tzrec/datasets/odps_dataset_v1.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 1
number of parameters: 0

id: 295
unit: def __init__()
file: tzrec/datasets/dataset.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 1
number of parameters: 0

id: 296
unit: def stop_char_file()
file: tzrec/features/tokenize_feature.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 4
number of parameters: 1

id: 297
unit: def __init__()
file: tzrec/features/sequence_feature.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 1
number of parameters: 0

id: 298
unit: def _build_side_inputs()
file: tzrec/features/sequence_feature.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 3
number of parameters: 1

id: 299
unit: def __init__()
file: tzrec/features/sequence_feature.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 1
number of parameters: 0

id: 300
unit: def _build_side_inputs()
file: tzrec/features/sequence_feature.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 3
number of parameters: 1

id: 301
unit: def __init__()
file: tzrec/features/sequence_feature.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 1
number of parameters: 0

id: 302
unit: def _get_benchmark_project()
file: tzrec/benchmark/benchmark.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 5
number of parameters: 0

id: 303
unit: def _get_valid_attn_mask()
file: tzrec/ops/pytorch/pt_hstu_attention.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 1
number of parameters: 0

id: 304
unit: def prev_power_of_2()
file: tzrec/ops/utils.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 3
number of parameters: 1

id: 305
unit: def hstu_concat_l2_embeddings()
file: tzrec/ops/jagged_tensors.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 1
number of parameters: 0

id: 306
unit: def _ln_mul_dropout_bwd_dwdb()
file: tzrec/ops/triton/triton_hstu_linear.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 1
number of parameters: 0

id: 307
unit: def forward()
file: tzrec/ops/triton/triton_jagged_tensors.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 1
number of parameters: 0

id: 308
unit: def forward()
file: tzrec/ops/triton/triton_jagged_tensors.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 1
number of parameters: 0

id: 309
unit: def _layer_norm_bwd_dwdb()
file: tzrec/ops/triton/triton_layer_norm.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 1
number of parameters: 0

id: 310
unit: def __init__()
file: tzrec/modules/sequence.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 1
number of parameters: 0

id: 311
unit: def __init__()
file: tzrec/modules/interaction.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 1
number of parameters: 2

id: 312
unit: def __init__()
file: tzrec/modules/dense_embedding_collection.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 1
number of parameters: 0

id: 313
unit: def group_total_dim()
file: tzrec/modules/embedding.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 3
number of parameters: 2

id: 314
unit: def group_total_dim()
file: tzrec/modules/embedding.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 2
number of parameters: 2

id: 315
unit: def _forward_impl()
file: tzrec/modules/embedding.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 1
number of parameters: 0

id: 316
unit: def forward()
file: tzrec/modules/intervention.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 1
number of parameters: 3

id: 317
unit: def sequence_mask()
file: tzrec/modules/capsule.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 2
number of parameters: 2

id: 318
unit: def __init__()
file: tzrec/models/multi_tower_din_trt.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 1
number of parameters: 0

id: 319
unit: def _update_metric_impl()
file: tzrec/models/rank_model.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 1
number of parameters: 0

id: 320
unit: def __init__()
file: tzrec/models/dssm_v2.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 1
number of parameters: 0

id: 321
unit: def __init__()
file: tzrec/models/mind.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 1
number of parameters: 0

id: 322
unit: def forward()
file: tzrec/models/mind.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 3
number of parameters: 2

id: 323
unit: def __init__()
file: tzrec/models/dat.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 1
number of parameters: 0

id: 324
unit: def __init__()
file: tzrec/models/dssm.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 1
number of parameters: 0

id: 325
unit: def forward()
file: tzrec/models/dssm.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 3
number of parameters: 2

id: 326
unit: def _init_loss_impl()
file: tzrec/models/match_model.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 2
number of parameters: 3

id: 327
unit: def save_serving_tree()
file: tzrec/tools/tdm/gen_tree/tree_search_util.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 4
number of parameters: 2

id: 328
unit: def save_tree()
file: tzrec/tools/tdm/gen_tree/tree_builder.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 1
number of parameters: 2

id: 329
unit: def __init__()
file: tzrec/datasets/data_parser.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 1
number of parameters: 0

id: 330
unit: def _get_cluster_spec()
file: tzrec/datasets/sampler.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 1
number of parameters: 1

id: 331
unit: def _get_compression_type()
file: tzrec/datasets/odps_dataset.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 3
number of parameters: 1

id: 332
unit: def _reader_iter()
file: tzrec/datasets/odps_dataset.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 1
number of parameters: 0

id: 333
unit: def _reader_iter()
file: tzrec/datasets/parquet_dataset.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 1
number of parameters: 0

id: 334
unit: def __init__()
file: tzrec/datasets/utils.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 2
number of parameters: 2

id: 335
unit: def get()
file: tzrec/datasets/utils.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 2
number of parameters: 1

id: 336
unit: def remove_nullable()
file: tzrec/datasets/utils.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 2
number of parameters: 1

id: 337
unit: def __init__()
file: tzrec/datasets/dataset.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 1
number of parameters: 0

id: 338
unit: def init_fg()
file: tzrec/features/feature.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 3
number of parameters: 1

id: 339
unit: def vocab_file()
file: tzrec/features/tokenize_feature.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 4
number of parameters: 1

id: 340
unit: def _get_dataloader()
file: tzrec/main.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 1
number of parameters: 0

id: 341
unit: def _evaluate()
file: tzrec/main.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 1
number of parameters: 0

id: 342
unit: def train_and_evaluate()
file: tzrec/main.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 1
number of parameters: 0

id: 343
unit: def _forward_loop()
file: tzrec/main.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 3
number of parameters: 0

id: 344
unit: def _flops()
file: tzrec/ops/benchmarks/hstu_attention_bench.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 1
number of parameters: 0

id: 345
unit: def pytorch_hstu_concat_l2_embeddings()
file: tzrec/ops/pytorch/pt_jagged_tensors.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 1
number of parameters: 0

id: 346
unit: def concat_2D_jagged()
file: tzrec/ops/jagged_tensors.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 1
number of parameters: 0

id: 347
unit: def split_2D_jagged()
file: tzrec/ops/jagged_tensors.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 1
number of parameters: 0

id: 348
unit: def triton_concat_2D_jagged()
file: tzrec/ops/triton/triton_jagged_tensors.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 1
number of parameters: 0

id: 349
unit: def triton_split_2D_jagged()
file: tzrec/ops/triton/triton_jagged_tensors.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 1
number of parameters: 0

id: 350
unit: def forward()
file: tzrec/ops/triton/triton_position.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 1
number of parameters: 0

id: 351
unit: def get_version()
file: setup.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 2
number of parameters: 0

id: 352
unit: def __init__()
file: tzrec/modules/sequence.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 0

id: 353
unit: def forward()
file: tzrec/modules/interaction.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 2

id: 354
unit: def __init__()
file: tzrec/modules/masknet_module.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 0

id: 355
unit: def __init__()
file: tzrec/modules/dense_embedding_collection.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 0

id: 356
unit: def __init__()
file: tzrec/modules/intervention.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 3

id: 357
unit: def __init__()
file: tzrec/modules/mmoe.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 0

id: 358
unit: def __init__()
file: tzrec/optim/lr_scheduler.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 0

id: 359
unit: def save_message()
file: tzrec/utils/config_util.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 2
number of parameters: 2

id: 360
unit: def __init__()
file: tzrec/models/multi_tower_din_trt.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 0

id: 361
unit: def __init__()
file: tzrec/models/dbmtl.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 0

id: 362
unit: def __init__()
file: tzrec/models/deepfm.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 0

id: 363
unit: def __init__()
file: tzrec/models/rank_model.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 0

id: 364
unit: def forward()
file: tzrec/models/dssm_v2.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 3
number of parameters: 2

id: 365
unit: def __init__()
file: tzrec/models/dssm_v2.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 0

id: 366
unit: def __init__()
file: tzrec/models/mind.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 0

id: 367
unit: def __init__()
file: tzrec/models/ple.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 0

id: 368
unit: def __init__()
file: tzrec/models/dat.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 0

id: 369
unit: def __init__()
file: tzrec/models/rocket_launching.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 0

id: 370
unit: def init_metric()
file: tzrec/models/rocket_launching.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 3
number of parameters: 1

id: 371
unit: def __init__()
file: tzrec/models/multi_tower_din.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 0

id: 372
unit: def __init__()
file: tzrec/models/tdm.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 0

id: 373
unit: def __init__()
file: tzrec/models/dssm.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 0

id: 374
unit: def __init__()
file: tzrec/models/mmoe.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 0

id: 375
unit: def __init__()
file: tzrec/models/multi_task_rank.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 0

id: 376
unit: def __init__()
file: tzrec/models/multi_tower.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 0

id: 377
unit: def __init__()
file: tzrec/models/masknet.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 0

id: 378
unit: def __init__()
file: tzrec/models/hstu.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 0

id: 379
unit: def __init__()
file: tzrec/models/hstu.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 0

id: 380
unit: def __init__()
file: tzrec/models/dlrm.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 0

id: 381
unit: def __init__()
file: tzrec/models/match_model.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 0

id: 382
unit: def _loss_impl()
file: tzrec/models/match_model.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 0

id: 383
unit: def _update_metric_impl()
file: tzrec/models/match_model.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 0

id: 384
unit: def __init__()
file: tzrec/models/match_model.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 3

id: 385
unit: def predict()
file: tzrec/models/match_model.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 2

id: 386
unit: def __init__()
file: tzrec/models/model.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 0

id: 387
unit: def _update_loss_metric_impl()
file: tzrec/models/model.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 0

id: 388
unit: def forward()
file: tzrec/models/model.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 3
number of parameters: 2

id: 389
unit: def __init__()
file: tzrec/tools/tdm/gen_tree/tree_search_util.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 0

id: 390
unit: def interest_merge()
file: tzrec/tools/hitrate.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 0

id: 391
unit: def __init__()
file: tzrec/tools/add_feature_info_to_config.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 0

id: 392
unit: def _init_fg_hander()
file: tzrec/datasets/data_parser.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 2
number of parameters: 1

id: 393
unit: def __init__()
file: tzrec/datasets/sampler.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 6

id: 394
unit: def __init__()
file: tzrec/datasets/sampler.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 6

id: 395
unit: def get()
file: tzrec/datasets/sampler.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 3

id: 396
unit: def __init__()
file: tzrec/datasets/sampler.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 6

id: 397
unit: def __init__()
file: tzrec/datasets/sampler.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 6

id: 398
unit: def __init__()
file: tzrec/datasets/sampler.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 6

id: 399
unit: def __init__()
file: tzrec/datasets/sampler.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 6

id: 400
unit: def __init__()
file: tzrec/datasets/sampler.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 6

id: 401
unit: def calc_slice_position()
file: tzrec/datasets/utils.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 0

id: 402
unit: def __iter__()
file: tzrec/datasets/dataset.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 4
number of parameters: 1

id: 403
unit: def _embedding_dim()
file: tzrec/features/feature.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 2
number of parameters: 1

id: 404
unit: def inputs()
file: tzrec/features/feature.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 4
number of parameters: 1

id: 405
unit: def inputs()
file: tzrec/features/id_feature.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 4
number of parameters: 1

id: 406
unit: def _parse_fg_encoded_sequence_dense_feature_impl()
file: tzrec/features/sequence_feature.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 4

id: 407
unit: def _write_loop()
file: tzrec/main.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 4
number of parameters: 1

id: 408
unit: def is_trt_predict()
file: tzrec/acc/utils.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 3
number of parameters: 1

id: 409
unit: def _get_kernel()
file: tzrec/ops/benchmarks/hstu_attention_bench.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 3
number of parameters: 1

id: 410
unit: def _concat_2D_jagged_jagged()
file: tzrec/ops/pytorch/pt_jagged_tensors.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 0

id: 411
unit: def pytorch_concat_2D_jagged()
file: tzrec/ops/pytorch/pt_jagged_tensors.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 0

id: 412
unit: def pytorch_split_2D_jagged()
file: tzrec/ops/pytorch/pt_jagged_tensors.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 0

id: 413
unit: def pytorch_add_position_embeddings()
file: tzrec/ops/pytorch/pt_position.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 0

id: 414
unit: def _get_col_indices()
file: tzrec/ops/pytorch/pt_position.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 0

id: 415
unit: def switch_to_contiguous_if_needed()
file: tzrec/ops/utils.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 4
number of parameters: 1

id: 416
unit: def hstu_split_l2_embeddings()
file: tzrec/ops/jagged_tensors.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 0

id: 417
unit: def jagged_dense_bmm_broadcast_add()
file: tzrec/ops/jagged_tensors.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 0

id: 418
unit: def _group_norm_bwd_dwdb()
file: tzrec/ops/triton/triton_hstu_linear.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 0

id: 419
unit: def forward()
file: tzrec/ops/triton/triton_jagged_tensors.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 0

id: 420
unit: def triton_add_position_embeddings()
file: tzrec/ops/triton/triton_position.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 0

id: 421
unit: def triton_weighted_layer_norm_fwd()
file: tzrec/ops/triton/triton_layer_norm.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 0

id: 422
unit: def _rms_norm_bwd_dwdb()
file: tzrec/ops/triton/triton_layer_norm.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 0

id: 423
unit: def __init__()
file: tzrec/modules/sequence.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 0

id: 424
unit: def __init__()
file: tzrec/modules/sequence.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 0

id: 425
unit: def forward()
file: tzrec/modules/masknet_module.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 3

id: 426
unit: def __init__()
file: tzrec/modules/embedding.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 0

id: 427
unit: def group_feature_dims()
file: tzrec/modules/embedding.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 3
number of parameters: 2

id: 428
unit: def __init__()
file: tzrec/modules/embedding.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 0

id: 429
unit: def forward()
file: tzrec/modules/embedding.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 0

id: 430
unit: def jagged_forward()
file: tzrec/modules/embedding.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 0

id: 431
unit: def __init__()
file: tzrec/modules/intervention.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 0

id: 432
unit: def forward()
file: tzrec/modules/fm.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 2

id: 433
unit: def __init__()
file: tzrec/modules/activation.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 0

id: 434
unit: def __init__()
file: tzrec/modules/variational_dropout.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 0

id: 435
unit: def kernel()
file: tzrec/modules/utils.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 2
number of parameters: 1

id: 436
unit: def div_no_nan()
file: tzrec/modules/utils.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 0

id: 437
unit: def _init_routing_logits()
file: tzrec/modules/capsule.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 2

id: 438
unit: def dynamic_routing()
file: tzrec/modules/capsule.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 0

id: 439
unit: def __init__()
file: tzrec/modules/hstu.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 2

id: 440
unit: def register_class()
file: tzrec/utils/load_class.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 2
number of parameters: 3

id: 441
unit: def feedback()
file: tzrec/utils/plan_util.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 0

id: 442
unit: def get_free_port()
file: tzrec/utils/misc_util.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 1

id: 443
unit: def _output_to_prediction_impl()
file: tzrec/models/rank_model.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 0

id: 444
unit: def _init_loss_impl()
file: tzrec/models/rank_model.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 0

id: 445
unit: def predict()
file: tzrec/models/masknet.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 2

id: 446
unit: def init_loss()
file: tzrec/models/match_model.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 2
number of parameters: 1

id: 447
unit: def compute_metric()
file: tzrec/models/model.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 2
number of parameters: 1

id: 448
unit: def _ancestors()
file: tzrec/tools/tdm/gen_tree/tree_builder.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 2
number of parameters: 2

id: 449
unit: def __init__()
file: tzrec/tools/convert_easyrec_config_to_tzrec_config.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 0

id: 450
unit: def load_easyrec_config()
file: tzrec/tools/convert_easyrec_config_to_tzrec_config.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 2

id: 451
unit: def _easyrec_dnn_2_tzrec_mlp()
file: tzrec/tools/convert_easyrec_config_to_tzrec_config.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 2

id: 452
unit: def _easyrec_tower_2_tzrec_tower()
file: tzrec/tools/convert_easyrec_config_to_tzrec_config.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 2

id: 453
unit: def _easyrec_dssm_tower_2_tzrec_tower()
file: tzrec/tools/convert_easyrec_config_to_tzrec_config.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 2

id: 454
unit: def batch_hitrate()
file: tzrec/tools/hitrate.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 0

id: 455
unit: def _pa_ids_to_npy()
file: tzrec/datasets/sampler.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 2
number of parameters: 1

id: 456
unit: def init()
file: tzrec/datasets/sampler.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 2

id: 457
unit: def init_sampler()
file: tzrec/datasets/sampler.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 2

id: 458
unit: def get()
file: tzrec/datasets/sampler.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 3

id: 459
unit: def __init__()
file: tzrec/datasets/odps_dataset.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 0

id: 460
unit: def _create_partition()
file: tzrec/datasets/odps_dataset.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 3
number of parameters: 1

id: 461
unit: def __init__()
file: tzrec/datasets/odps_dataset_v1.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 0

id: 462
unit: def __init__()
file: tzrec/datasets/parquet_dataset.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 0

id: 463
unit: def process_hstu_neg_sample()
file: tzrec/datasets/utils.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 0

id: 464
unit: def create_reader()
file: tzrec/datasets/dataset.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 0

id: 465
unit: def __init__()
file: tzrec/datasets/csv_dataset.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 0

id: 466
unit: def _parse_fg_encoded_sparse_feature_impl()
file: tzrec/features/feature.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 3

id: 467
unit: def create_features()
file: tzrec/features/feature.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 0

id: 468
unit: def _parse_fg_encoded_sequence_sparse_feature_impl()
file: tzrec/features/sequence_feature.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 4

id: 469
unit: def _log_train()
file: tzrec/main.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 0

id: 470
unit: def _script_model()
file: tzrec/main.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 0

id: 471
unit: def is_aot()
file: tzrec/acc/utils.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 3
number of parameters: 0

id: 472
unit: def _print()
file: tzrec/benchmark/benchmark.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 0

id: 473
unit: def _pad_qkv()
file: tzrec/ops/pytorch/pt_hstu_attention.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 0

id: 474
unit: def pytorch_layer_norm()
file: tzrec/ops/pytorch/pt_layer_norm.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 0

id: 475
unit: def pytorch_swish_layer_norm()
file: tzrec/ops/pytorch/pt_layer_norm.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 0

id: 476
unit: def pytorch_hstu_split_l2_embeddings()
file: tzrec/ops/pytorch/pt_jagged_tensors.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 0

id: 477
unit: def pytorch_jagged_dense_bmm_broadcast_add()
file: tzrec/ops/pytorch/pt_jagged_tensors.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 0

id: 478
unit: def layer_norm()
file: tzrec/ops/layer_norm.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 0

id: 479
unit: def swish_layer_norm()
file: tzrec/ops/layer_norm.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 0

id: 480
unit: def triton_jagged_dense_bmm_broadcast_add()
file: tzrec/ops/triton/triton_jagged_tensors.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 0

id: 481
unit: def forward()
file: tzrec/ops/triton/triton_layer_norm.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 0

id: 482
unit: def forward()
file: tzrec/ops/triton/triton_layer_norm.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 0

id: 483
unit: def triton_swish_layer_norm()
file: tzrec/ops/triton/triton_layer_norm.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 0

id: 484
unit: def update()
file: tzrec/metrics/recall_at_k.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 3

id: 485
unit: def output_dim()
file: tzrec/modules/interaction.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 2
number of parameters: 1

id: 486
unit: def _gate_forward()
file: tzrec/modules/extraction_net.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 0

id: 487
unit: def __init__()
file: tzrec/modules/masknet_module.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 0

id: 488
unit: def __init__()
file: tzrec/modules/dense_embedding_collection.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 0

id: 489
unit: def group_key()
file: tzrec/modules/dense_embedding_collection.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 1

id: 490
unit: def __init__()
file: tzrec/modules/dense_embedding_collection.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 0

id: 491
unit: def __init__()
file: tzrec/modules/dense_embedding_collection.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 0

id: 492
unit: def __init__()
file: tzrec/modules/embedding.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 0

id: 493
unit: def __init__()
file: tzrec/modules/embedding.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 0

id: 494
unit: def __init__()
file: tzrec/modules/task_tower.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 0

id: 495
unit: def forward()
file: tzrec/modules/task_tower.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 2
number of parameters: 2

id: 496
unit: def __init__()
file: tzrec/modules/hstu.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 0

id: 497
unit: def create_scheduler()
file: tzrec/optim/optimizer_builder.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 0

id: 498
unit: def __init__()
file: tzrec/utils/checkpoint_util.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 0

id: 499
unit: def restore_model()
file: tzrec/utils/checkpoint_util.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 0

id: 500
unit: def random_name()
file: tzrec/utils/misc_util.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 2
number of parameters: 1

id: 501
unit: def _forward_impl()
file: tzrec/utils/fx_util.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 0

id: 502
unit: def init_metric()
file: tzrec/models/rank_model.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 3
number of parameters: 1

id: 503
unit: def update_metric()
file: tzrec/models/rank_model.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 0

id: 504
unit: def label_aware_attention()
file: tzrec/models/mind.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 0

id: 505
unit: def feature_based_sim()
file: tzrec/models/rocket_launching.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 0

id: 506
unit: def update_metric()
file: tzrec/models/rocket_launching.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 0

id: 507
unit: def update_metric()
file: tzrec/models/multi_task_rank.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 0

id: 508
unit: def sim()
file: tzrec/models/hstu.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 0

id: 509
unit: def init_metric()
file: tzrec/models/match_model.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 3
number of parameters: 1

id: 510
unit: def update_metric()
file: tzrec/models/match_model.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 0

id: 511
unit: def __init__()
file: tzrec/models/match_model.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 3

id: 512
unit: def update_metric()
file: tzrec/models/model.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 0

id: 513
unit: def __init__()
file: tzrec/models/model.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 2

id: 514
unit: def __init__()
file: tzrec/models/model.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 2

id: 515
unit: def _forward()
file: tzrec/tools/tdm/retrieval.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 0

id: 516
unit: def _iter()
file: tzrec/tools/tdm/gen_tree/tree_search_util.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 0

id: 517
unit: def _load()
file: tzrec/tools/tdm/gen_tree/tree_search_util.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 2

id: 518
unit: def _create_train_config()
file: tzrec/tools/convert_easyrec_config_to_tzrec_config.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 2
number of parameters: 2

id: 519
unit: def _create_eval_config()
file: tzrec/tools/convert_easyrec_config_to_tzrec_config.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 2
number of parameters: 2

id: 520
unit: def init_cluster()
file: tzrec/datasets/sampler.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 0

id: 521
unit: def launch_server()
file: tzrec/datasets/sampler.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 2
number of parameters: 1

id: 522
unit: def _get_metadata()
file: tzrec/datasets/parquet_dataset.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 1

id: 523
unit: def _expand_tdm_sample()
file: tzrec/datasets/dataset.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 0

id: 524
unit: def launch_sampler_cluster()
file: tzrec/datasets/dataset.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 0

id: 525
unit: def sampled_batch_size()
file: tzrec/datasets/dataset.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 2
number of parameters: 1

id: 526
unit: def __init__()
file: tzrec/features/expr_feature.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 0

id: 527
unit: def output_dim()
file: tzrec/features/expr_feature.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 2
number of parameters: 1

id: 528
unit: def _build_side_inputs()
file: tzrec/features/expr_feature.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 3
number of parameters: 1

id: 529
unit: def __init__()
file: tzrec/features/lookup_feature.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 0

id: 530
unit: def value_dim()
file: tzrec/features/lookup_feature.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 2
number of parameters: 1

id: 531
unit: def output_dim()
file: tzrec/features/lookup_feature.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 2
number of parameters: 1

id: 532
unit: def _build_side_inputs()
file: tzrec/features/lookup_feature.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 4
number of parameters: 1

id: 533
unit: def assets()
file: tzrec/features/lookup_feature.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 2
number of parameters: 1

id: 534
unit: def __init__()
file: tzrec/features/overlap_feature.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 0

id: 535
unit: def output_dim()
file: tzrec/features/overlap_feature.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 2
number of parameters: 1

id: 536
unit: def _build_side_inputs()
file: tzrec/features/overlap_feature.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 4
number of parameters: 1

id: 537
unit: def _parse_fg_encoded_dense_feature_impl()
file: tzrec/features/feature.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 3

id: 538
unit: def __init__()
file: tzrec/features/feature.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 0

id: 539
unit: def has_embedding()
file: tzrec/features/feature.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 2
number of parameters: 1

id: 540
unit: def default_bucketize_value()
file: tzrec/features/feature.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 2
number of parameters: 1

id: 541
unit: def __init__()
file: tzrec/features/custom_feature.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 0

id: 542
unit: def value_dim()
file: tzrec/features/custom_feature.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 2
number of parameters: 1

id: 543
unit: def output_dim()
file: tzrec/features/custom_feature.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 2
number of parameters: 1

id: 544
unit: def _build_side_inputs()
file: tzrec/features/custom_feature.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 3
number of parameters: 1

id: 545
unit: def __init__()
file: tzrec/features/raw_feature.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 0

id: 546
unit: def value_dim()
file: tzrec/features/raw_feature.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 2
number of parameters: 1

id: 547
unit: def output_dim()
file: tzrec/features/raw_feature.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 2
number of parameters: 1

id: 548
unit: def _build_side_inputs()
file: tzrec/features/raw_feature.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 2
number of parameters: 1

id: 549
unit: def __init__()
file: tzrec/features/id_feature.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 0

id: 550
unit: def value_dim()
file: tzrec/features/id_feature.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 2
number of parameters: 1

id: 551
unit: def _build_side_inputs()
file: tzrec/features/id_feature.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 2
number of parameters: 1

id: 552
unit: def assets()
file: tzrec/features/id_feature.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 2
number of parameters: 1

id: 553
unit: def __init__()
file: tzrec/features/tokenize_feature.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 0

id: 554
unit: def assets()
file: tzrec/features/tokenize_feature.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 2
number of parameters: 1

id: 555
unit: def __init__()
file: tzrec/features/match_feature.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 0

id: 556
unit: def value_dim()
file: tzrec/features/match_feature.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 2
number of parameters: 1

id: 557
unit: def output_dim()
file: tzrec/features/match_feature.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 2
number of parameters: 1

id: 558
unit: def assets()
file: tzrec/features/match_feature.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 2
number of parameters: 1

id: 559
unit: def __init__()
file: tzrec/features/combo_feature.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 0

id: 560
unit: def _build_side_inputs()
file: tzrec/features/combo_feature.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 3
number of parameters: 1

id: 561
unit: def name()
file: tzrec/features/sequence_feature.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 2
number of parameters: 1

id: 562
unit: def value_dim()
file: tzrec/features/sequence_feature.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 2
number of parameters: 1

id: 563
unit: def name()
file: tzrec/features/sequence_feature.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 2
number of parameters: 1

id: 564
unit: def name()
file: tzrec/features/sequence_feature.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 2
number of parameters: 1

id: 565
unit: def _create_model()
file: tzrec/main.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 0

id: 566
unit: def evaluate()
file: tzrec/main.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 0

id: 567
unit: def export()
file: tzrec/main.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 0

id: 568
unit: def is_input_tile()
file: tzrec/acc/utils.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 4
number of parameters: 0

id: 569
unit: def is_input_tile_emb()
file: tzrec/acc/utils.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 3
number of parameters: 0

id: 570
unit: def is_trt()
file: tzrec/acc/utils.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 3
number of parameters: 0

id: 571
unit: def is_debug_trt()
file: tzrec/acc/utils.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 3
number of parameters: 0

id: 572
unit: def is_quant()
file: tzrec/acc/utils.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 3
number of parameters: 0

id: 573
unit: def _quantize_state_dict()
file: tzrec/acc/utils.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 0

id: 574
unit: def _get_train_metrics()
file: tzrec/benchmark/benchmark.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 1

id: 575
unit: def _split_2D_jagged_jagged()
file: tzrec/ops/pytorch/pt_jagged_tensors.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 0

id: 576
unit: def _get_high_inds()
file: tzrec/ops/position.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 0

id: 577
unit: def addmm()
file: tzrec/ops/mm.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 0

id: 578
unit: def triton_addmm_bwd()
file: tzrec/ops/triton/triton_addmm.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 0

id: 579
unit: def forward()
file: tzrec/ops/triton/triton_addmm.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 0

id: 580
unit: def forward()
file: tzrec/ops/triton/triton_layer_norm.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 0

id: 581
unit: def triton_layer_norm()
file: tzrec/ops/triton/triton_layer_norm.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 0

id: 582
unit: def forward()
file: tzrec/loss/jrc_loss.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 0

id: 583
unit: def readme()
file: setup.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 0

id: 584
unit: def get_current_embeddings()
file: tzrec/modules/sequence.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 0

id: 585
unit: def forward()
file: tzrec/modules/extraction_net.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 0

id: 586
unit: def group_key()
file: tzrec/modules/dense_embedding_collection.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 1

id: 587
unit: def forward()
file: tzrec/modules/dense_embedding_collection.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 2

id: 588
unit: def _add_mc_module()
file: tzrec/modules/embedding.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 0

id: 589
unit: def group_dims()
file: tzrec/modules/embedding.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 2
number of parameters: 2

id: 590
unit: def group_total_dim()
file: tzrec/modules/embedding.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 2
number of parameters: 2

id: 591
unit: def __init__()
file: tzrec/modules/utils.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 0

id: 592
unit: def recursive_setattr()
file: tzrec/modules/utils.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 3
number of parameters: 3

id: 593
unit: def __init__()
file: tzrec/modules/utils.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 3

id: 594
unit: def _norm_attn_output()
file: tzrec/modules/hstu.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 2

id: 595
unit: def config_to_kwargs()
file: tzrec/utils/config_util.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 1

id: 596
unit: def load()
file: tzrec/utils/plan_util.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 0

id: 597
unit: def _reset()
file: tzrec/utils/plan_util.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 1

id: 598
unit: def _mc_input_dist()
file: tzrec/utils/dist_util.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 0

id: 599
unit: def symbolic_trace()
file: tzrec/utils/fx_util.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 0

id: 600
unit: def predict()
file: tzrec/models/multi_tower_din_trt.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 2

id: 601
unit: def init_loss()
file: tzrec/models/rank_model.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 3
number of parameters: 1

id: 602
unit: def predict()
file: tzrec/models/tdm.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 2

id: 603
unit: def forward()
file: tzrec/models/hstu.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 3

id: 604
unit: def _init_loss_metric_impl()
file: tzrec/models/model.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 3

id: 605
unit: def get_batch()
file: tzrec/models/model.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 0

id: 606
unit: def forward()
file: tzrec/models/model.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 0

id: 607
unit: def generate()
file: tzrec/tools/tdm/gen_tree/tree_generator.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 2

id: 608
unit: def _add_suffix_to_odps_table()
file: tzrec/tools/tdm/gen_tree/tree_search_util.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 2

id: 609
unit: def build()
file: tzrec/tools/tdm/gen_tree/tree_builder.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 0

id: 610
unit: def load_easyrec_fg_json()
file: tzrec/tools/convert_easyrec_config_to_tzrec_config.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 2

id: 611
unit: def _update_feature_group()
file: tzrec/tools/add_feature_info_to_config.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 0

id: 612
unit: def _to_tensor()
file: tzrec/datasets/data_parser.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 2
number of parameters: 1

id: 613
unit: def estimated_sample_num()
file: tzrec/datasets/sampler.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 1

id: 614
unit: def _pa_read()
file: tzrec/datasets/odps_dataset_v1.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 0

id: 615
unit: def close()
file: tzrec/datasets/parquet_dataset.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 2
number of parameters: 1

id: 616
unit: def process_hstu_seq_data()
file: tzrec/datasets/utils.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 0

id: 617
unit: def input_fields()
file: tzrec/datasets/dataset.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 2
number of parameters: 1

id: 618
unit: def close()
file: tzrec/datasets/csv_dataset.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 2
number of parameters: 1

id: 619
unit: def feature_config()
file: tzrec/features/feature.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 2

id: 620
unit: def is_sparse()
file: tzrec/features/feature.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 2
number of parameters: 1

id: 621
unit: def pooling_type()
file: tzrec/features/feature.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 1

id: 622
unit: def _copy_assets()
file: tzrec/features/feature.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 0

id: 623
unit: def assets()
file: tzrec/features/custom_feature.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 1

id: 624
unit: def is_sparse()
file: tzrec/features/raw_feature.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 2
number of parameters: 1

id: 625
unit: def is_sparse()
file: tzrec/features/id_feature.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 2
number of parameters: 1

id: 626
unit: def _write()
file: tzrec/main.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 0

id: 627
unit: def forward()
file: tzrec/acc/trt_utils.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 0

id: 628
unit: def __init__()
file: tzrec/acc/trt_utils.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 3

id: 629
unit: def forward()
file: tzrec/acc/trt_utils.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 0

id: 630
unit: def _modify_pipline_config()
file: tzrec/benchmark/benchmark.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 0

id: 631
unit: def _create_directory()
file: tzrec/benchmark/benchmark.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 2
number of parameters: 1

id: 632
unit: def main()
file: tzrec/benchmark/benchmark.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 0

id: 633
unit: def _apply_sampling()
file: tzrec/ops/benchmarks/hstu_attention_bench.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 0

id: 634
unit: def set_static_max_seq_lens()
file: tzrec/ops/utils.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 1

id: 635
unit: def _bwd_pre_hook()
file: tzrec/ops/triton/triton_hstu_attention.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 2
number of parameters: 1

id: 636
unit: def backward()
file: tzrec/ops/triton/triton_hstu_preprocess_and_attention.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 0

id: 637
unit: def triton_addmm_fwd()
file: tzrec/ops/triton/triton_addmm.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 0

id: 638
unit: def triton_addmm()
file: tzrec/ops/triton/triton_addmm.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 0

id: 639
unit: def triton_rms_norm()
file: tzrec/ops/triton/triton_layer_norm.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 0

id: 640
unit: def __init__()
file: tzrec/loss/jrc_loss.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 0

id: 641
unit: def __init__()
file: tzrec/metrics/grouped_auc.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 2

id: 642
unit: def __init__()
file: tzrec/modules/sequence.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 2

id: 643
unit: def has_group()
file: tzrec/modules/embedding.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 2
number of parameters: 2

id: 644
unit: def forward()
file: tzrec/modules/embedding.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 0

id: 645
unit: def predict()
file: tzrec/modules/embedding.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 0

id: 646
unit: def has_group()
file: tzrec/modules/embedding.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 2
number of parameters: 2

id: 647
unit: def forward()
file: tzrec/modules/embedding.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 0

id: 648
unit: def jagged_forward()
file: tzrec/modules/embedding.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 0

id: 649
unit: def has_group()
file: tzrec/modules/embedding.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 2
number of parameters: 2

id: 650
unit: def _feature_tile()
file: tzrec/modules/variational_dropout.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 0

id: 651
unit: def set_is_inference()
file: tzrec/modules/utils.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 2

id: 652
unit: def set_kernel()
file: tzrec/modules/utils.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 2

id: 653
unit: def forward()
file: tzrec/modules/hstu.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 0

id: 654
unit: def forward()
file: tzrec/modules/hstu.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 0

id: 655
unit: def forward()
file: tzrec/modules/hstu.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 0

id: 656
unit: def __init__()
file: tzrec/optim/lr_scheduler.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 3

id: 657
unit: def fx_unwrap_optional_tensor()
file: tzrec/utils/fx_util.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 1

id: 658
unit: def _update_dict_tensor()
file: tzrec/models/dat.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 0

id: 659
unit: def _update_dict_tensor()
file: tzrec/models/dssm.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 0

id: 660
unit: def _update_dict_tensor()
file: tzrec/models/hstu.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 0

id: 661
unit: def forward()
file: tzrec/models/hstu.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 3

id: 662
unit: def forward()
file: tzrec/models/model.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 0

id: 663
unit: def __init__()
file: tzrec/tools/tdm/gen_tree/tree_builder.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 3

id: 664
unit: def __del__()
file: tzrec/datasets/data_parser.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 2
number of parameters: 1

id: 665
unit: def string_attrs()
file: tzrec/datasets/sampler.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 1

id: 666
unit: def string_attrs()
file: tzrec/datasets/sampler.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 2

id: 667
unit: def __del__()
file: tzrec/datasets/sampler.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 2
number of parameters: 1

id: 668
unit: def _read_rows_arrow_with_retry()
file: tzrec/datasets/odps_dataset.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 0

id: 669
unit: def get_sampler_cluster()
file: tzrec/datasets/dataset.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 2
number of parameters: 1

id: 670
unit: def __init__()
file: tzrec/datasets/dataset.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 3

id: 671
unit: def __del__()
file: tzrec/datasets/dataset.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 2
number of parameters: 1

id: 672
unit: def is_neg()
file: tzrec/features/expr_feature.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 2

id: 673
unit: def is_neg()
file: tzrec/features/lookup_feature.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 2

id: 674
unit: def is_neg()
file: tzrec/features/overlap_feature.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 2

id: 675
unit: def is_neg()
file: tzrec/features/feature.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 2

id: 676
unit: def __del__()
file: tzrec/features/feature.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 3
number of parameters: 1

id: 677
unit: def is_neg()
file: tzrec/features/custom_feature.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 2

id: 678
unit: def is_neg()
file: tzrec/features/match_feature.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 2

id: 679
unit: def is_neg()
file: tzrec/features/combo_feature.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 2

id: 680
unit: def trt_convert()
file: tzrec/acc/trt_utils.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 0

id: 681
unit: def _benchmark_train_eval()
file: tzrec/benchmark/benchmark.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 0

id: 682
unit: def set_use_runtime_max_seq_len()
file: tzrec/ops/utils.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 1

id: 683
unit: def compute()
file: tzrec/metrics/recall_at_k.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 684
unit: def custom_reduce_fx()
file: tzrec/metrics/grouped_auc.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 685
unit: def update()
file: tzrec/metrics/grouped_auc.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 686
unit: def input()
file: tzrec/modules/sequence.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 687
unit: def output_dim()
file: tzrec/modules/sequence.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 688
unit: def output_dim()
file: tzrec/modules/sequence.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 689
unit: def output_dim()
file: tzrec/modules/sequence.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 690
unit: def output_dim()
file: tzrec/modules/sequence.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 691
unit: def output_dim()
file: tzrec/modules/sequence.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 692
unit: def output_dim()
file: tzrec/modules/sequence.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 693
unit: def create_seq_encoder()
file: tzrec/modules/sequence.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 694
unit: def _new_length_tensor()
file: tzrec/modules/interaction.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 695
unit: def forward()
file: tzrec/modules/interaction.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 696
unit: def output_dim()
file: tzrec/modules/extraction_net.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 697
unit: def _experts_layer_forward()
file: tzrec/modules/extraction_net.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 698
unit: def __init__()
file: tzrec/modules/dense_embedding_collection.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 3

id: 699
unit: def group_key()
file: tzrec/modules/dense_embedding_collection.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 700
unit: def reset_parameters()
file: tzrec/modules/dense_embedding_collection.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 701
unit: def merge_same_config_features()
file: tzrec/modules/dense_embedding_collection.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 702
unit: def _update_dict_tensor()
file: tzrec/modules/embedding.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 703
unit: def _merge_list_of_tensor_dict()
file: tzrec/modules/embedding.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 704
unit: def _merge_list_of_dict_of_jt_dict()
file: tzrec/modules/embedding.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 705
unit: def _merge_list_of_jt_dict()
file: tzrec/modules/embedding.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 706
unit: def _tile_and_combine_dense_kt()
file: tzrec/modules/embedding.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 707
unit: def grouped_features_keys()
file: tzrec/modules/embedding.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 708
unit: def _inspect_and_supplement_feature_group()
file: tzrec/modules/embedding.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 709
unit: def group_names()
file: tzrec/modules/embedding.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 710
unit: def _add_embedding_bag_config()
file: tzrec/modules/embedding.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 711
unit: def _add_embedding_config()
file: tzrec/modules/embedding.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 712
unit: def group_dims()
file: tzrec/modules/embedding.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 713
unit: def group_feature_dims()
file: tzrec/modules/embedding.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 714
unit: def group_total_dim()
file: tzrec/modules/embedding.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 715
unit: def group_names()
file: tzrec/modules/embedding.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 716
unit: def group_dims()
file: tzrec/modules/embedding.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 717
unit: def all_group_total_dim()
file: tzrec/modules/embedding.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 718
unit: def forward()
file: tzrec/modules/intervention.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 719
unit: def output_dim()
file: tzrec/modules/intervention.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 720
unit: def __init__()
file: tzrec/modules/fm.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 721
unit: def reset_parameters()
file: tzrec/modules/activation.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 722
unit: def forward()
file: tzrec/modules/mlp.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 723
unit: def output_dim()
file: tzrec/modules/mlp.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 724
unit: def forward()
file: tzrec/modules/mlp.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 725
unit: def _update_dict_tensor()
file: tzrec/modules/variational_dropout.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 726
unit: def concrete_dropout_neuron()
file: tzrec/modules/variational_dropout.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 727
unit: def output_dim()
file: tzrec/modules/mmoe.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 728
unit: def is_inference()
file: tzrec/modules/utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 729
unit: def is_eval()
file: tzrec/modules/utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 2
number of parameters: 1

id: 730
unit: def is_train()
file: tzrec/modules/utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 2
number of parameters: 1

id: 731
unit: def forward()
file: tzrec/modules/utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 732
unit: def __init__()
file: tzrec/modules/capsule.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 733
unit: def forward()
file: tzrec/modules/capsule.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 734
unit: def _norm_input()
file: tzrec/modules/hstu.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 735
unit: def by_epoch()
file: tzrec/optim/lr_scheduler.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 736
unit: def __init__()
file: tzrec/optim/lr_scheduler.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 737
unit: def get_lr()
file: tzrec/optim/lr_scheduler.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 738
unit: def create_sparse_optimizer()
file: tzrec/optim/optimizer_builder.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 739
unit: def create_dense_optimizer()
file: tzrec/optim/optimizer_builder.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 740
unit: def save_model()
file: tzrec/utils/checkpoint_util.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 741
unit: def load_pipeline_config()
file: tzrec/utils/config_util.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 742
unit: def which_msg()
file: tzrec/utils/config_util.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 743
unit: def use_hash_node_id()
file: tzrec/utils/env_util.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 744
unit: def _bytes_to_float_bin()
file: tzrec/utils/plan_util.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 3

id: 745
unit: def __init__()
file: tzrec/utils/logging_util.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 746
unit: def set_description()
file: tzrec/utils/logging_util.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 747
unit: def write_faiss_index()
file: tzrec/utils/faiss_util.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 748
unit: def search_faiss_index()
file: tzrec/utils/faiss_util.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 749
unit: def fx_arange()
file: tzrec/utils/fx_util.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 750
unit: def fx_int_item()
file: tzrec/utils/fx_util.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 751
unit: def _get_dict()
file: tzrec/models/multi_tower_din_trt.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 752
unit: def forward()
file: tzrec/models/multi_tower_din_trt.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 753
unit: def _update_tensor_dict()
file: tzrec/models/rank_model.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 754
unit: def _output_to_prediction()
file: tzrec/models/rank_model.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 755
unit: def loss()
file: tzrec/models/rank_model.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 756
unit: def _init_metric_impl()
file: tzrec/models/rank_model.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 757
unit: def forward()
file: tzrec/models/mind.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 758
unit: def __init__()
file: tzrec/models/dc2vr.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 759
unit: def forward()
file: tzrec/models/dat.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 760
unit: def loss()
file: tzrec/models/dat.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 761
unit: def _distillation_loss()
file: tzrec/models/rocket_launching.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 762
unit: def loss()
file: tzrec/models/rocket_launching.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 763
unit: def __init__()
file: tzrec/models/tdm.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 764
unit: def _multi_task_output_to_prediction()
file: tzrec/models/multi_task_rank.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 765
unit: def has_weight()
file: tzrec/models/multi_task_rank.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 766
unit: def loss()
file: tzrec/models/multi_task_rank.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 767
unit: def _zero_int_label()
file: tzrec/models/match_model.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 768
unit: def _arange_int_label()
file: tzrec/models/match_model.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 769
unit: def _update_tensor_2_dict()
file: tzrec/models/match_model.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 770
unit: def loss()
file: tzrec/models/match_model.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 771
unit: def predict()
file: tzrec/models/match_model.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 772
unit: def predict()
file: tzrec/models/model.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 773
unit: def init_loss()
file: tzrec/models/model.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 774
unit: def loss()
file: tzrec/models/model.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 775
unit: def init_metric()
file: tzrec/models/model.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 776
unit: def forward()
file: tzrec/models/model.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 777
unit: def get_features_in_feature_groups()
file: tzrec/models/model.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 778
unit: def update_data()
file: tzrec/tools/tdm/retrieval.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 779
unit: def _mini_batch()
file: tzrec/tools/tdm/gen_tree/tree_cluster.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 780
unit: def save_node_feature()
file: tzrec/tools/tdm/gen_tree/tree_search_util.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 781
unit: def set_parent()
file: tzrec/tools/tdm/gen_tree/tree_builder.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 782
unit: def set_children()
file: tzrec/tools/tdm/gen_tree/tree_builder.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 783
unit: def _easyrec_extraction_network_2_tzrec_extraction_network()
file: tzrec/tools/convert_easyrec_config_to_tzrec_config.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 784
unit: def _dump_to_csv()
file: tzrec/tools/feature_selection.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 785
unit: def _visualize_feature_importance()
file: tzrec/tools/feature_selection.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 786
unit: def _process_config()
file: tzrec/tools/feature_selection.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 787
unit: def _drop_feature_config()
file: tzrec/tools/add_feature_info_to_config.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 788
unit: def _update_feature_config()
file: tzrec/tools/add_feature_info_to_config.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 789
unit: def _parse_feature_normal()
file: tzrec/datasets/data_parser.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 790
unit: def _parse_feature_fg_handler()
file: tzrec/datasets/data_parser.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 791
unit: def to_batch()
file: tzrec/datasets/data_parser.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 792
unit: def _to_dense_features()
file: tzrec/datasets/data_parser.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 793
unit: def _to_sparse_features()
file: tzrec/datasets/data_parser.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 794
unit: def _to_dense_features_user1_itemb()
file: tzrec/datasets/data_parser.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 795
unit: def _to_sparse_features_user1tile_itemb()
file: tzrec/datasets/data_parser.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 796
unit: def _to_sparse_features_user1_itemb()
file: tzrec/datasets/data_parser.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 797
unit: def _to_arrow_array()
file: tzrec/datasets/sampler.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 3

id: 798
unit: def _parse_sparse_nodes()
file: tzrec/datasets/sampler.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 799
unit: def estimated_sample_num()
file: tzrec/datasets/sampler.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 800
unit: def estimated_sample_num()
file: tzrec/datasets/sampler.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 801
unit: def estimated_sample_num()
file: tzrec/datasets/sampler.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 802
unit: def estimated_sample_num()
file: tzrec/datasets/sampler.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 803
unit: def estimated_sample_num()
file: tzrec/datasets/sampler.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 804
unit: def estimated_sample_num()
file: tzrec/datasets/sampler.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 805
unit: def _iter_one_table()
file: tzrec/datasets/odps_dataset.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 806
unit: def to_batches()
file: tzrec/datasets/odps_dataset.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 807
unit: def __init__()
file: tzrec/datasets/odps_dataset.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 808
unit: def _iter_one_table()
file: tzrec/datasets/odps_dataset_v1.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 809
unit: def to_batches()
file: tzrec/datasets/odps_dataset_v1.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 810
unit: def to_batches()
file: tzrec/datasets/parquet_dataset.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 811
unit: def to_dict()
file: tzrec/datasets/utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 812
unit: def to_batches()
file: tzrec/datasets/dataset.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 813
unit: def _arrow_reader_iter()
file: tzrec/datasets/dataset.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 814
unit: def write()
file: tzrec/datasets/dataset.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 3

id: 815
unit: def close()
file: tzrec/datasets/dataset.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 816
unit: def create_writer()
file: tzrec/datasets/dataset.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 817
unit: def to_batches()
file: tzrec/datasets/csv_dataset.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 818
unit: def num_files()
file: tzrec/datasets/csv_dataset.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 819
unit: def value_dim()
file: tzrec/features/expr_feature.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 820
unit: def name()
file: tzrec/features/lookup_feature.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 821
unit: def _dense_emb_type()
file: tzrec/features/lookup_feature.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 822
unit: def value_dim()
file: tzrec/features/overlap_feature.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 823
unit: def name()
file: tzrec/features/feature.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 824
unit: def is_neg()
file: tzrec/features/feature.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 825
unit: def data_group()
file: tzrec/features/feature.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 826
unit: def data_group()
file: tzrec/features/feature.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 827
unit: def feature_config()
file: tzrec/features/feature.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 828
unit: def is_user_feat()
file: tzrec/features/feature.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 829
unit: def value_dim()
file: tzrec/features/feature.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 830
unit: def output_dim()
file: tzrec/features/feature.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 831
unit: def is_sequence()
file: tzrec/features/feature.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 832
unit: def is_grouped_sequence()
file: tzrec/features/feature.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 833
unit: def is_weighted()
file: tzrec/features/feature.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 834
unit: def num_embeddings()
file: tzrec/features/feature.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 835
unit: def _dense_emb_type()
file: tzrec/features/feature.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 836
unit: def dense_emb_config()
file: tzrec/features/feature.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 837
unit: def _build_side_inputs()
file: tzrec/features/feature.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 838
unit: def _parse()
file: tzrec/features/feature.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 3

id: 839
unit: def parse()
file: tzrec/features/feature.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 840
unit: def fg_json()
file: tzrec/features/feature.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 841
unit: def assets()
file: tzrec/features/feature.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 842
unit: def create_fg_json()
file: tzrec/features/feature.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 843
unit: def create_feature_configs()
file: tzrec/features/feature.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 844
unit: def name()
file: tzrec/features/custom_feature.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 845
unit: def _dense_emb_type()
file: tzrec/features/custom_feature.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 846
unit: def name()
file: tzrec/features/raw_feature.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 847
unit: def num_embeddings()
file: tzrec/features/raw_feature.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 848
unit: def _dense_emb_type()
file: tzrec/features/raw_feature.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 849
unit: def name()
file: tzrec/features/id_feature.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 850
unit: def output_dim()
file: tzrec/features/id_feature.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 851
unit: def value_dim()
file: tzrec/features/tokenize_feature.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 852
unit: def name()
file: tzrec/features/match_feature.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 853
unit: def _dense_emb_type()
file: tzrec/features/match_feature.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 854
unit: def is_sequence()
file: tzrec/features/sequence_feature.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 855
unit: def is_grouped_sequence()
file: tzrec/features/sequence_feature.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 856
unit: def is_sequence()
file: tzrec/features/sequence_feature.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 857
unit: def is_grouped_sequence()
file: tzrec/features/sequence_feature.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 858
unit: def _dense_emb_type()
file: tzrec/features/sequence_feature.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 2
number of parameters: 1

id: 859
unit: def is_sequence()
file: tzrec/features/sequence_feature.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 860
unit: def is_grouped_sequence()
file: tzrec/features/sequence_feature.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 861
unit: def _dense_emb_type()
file: tzrec/features/sequence_feature.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 862
unit: def _create_features()
file: tzrec/main.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 863
unit: def __init__()
file: tzrec/acc/trt_utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 864
unit: def get_trt_max_batch_size()
file: tzrec/acc/trt_utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 865
unit: def get_trt_max_seq_len()
file: tzrec/acc/trt_utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 866
unit: def export_model_trt()
file: tzrec/acc/trt_utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 867
unit: def export_pm()
file: tzrec/acc/export_utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 868
unit: def export_model_aot()
file: tzrec/acc/aot_utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 869
unit: def is_cuda_export()
file: tzrec/acc/utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 2
number of parameters: 0

id: 870
unit: def write_mapping_file_for_input_tile()
file: tzrec/acc/utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 871
unit: def print_error()
file: tzrec/benchmark/benchmark.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 872
unit: def print_worse()
file: tzrec/benchmark/benchmark.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 873
unit: def print_better()
file: tzrec/benchmark/benchmark.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 874
unit: def print_some_better_and_worse()
file: tzrec/benchmark/benchmark.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 875
unit: def print_balance()
file: tzrec/benchmark/benchmark.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 876
unit: def _compare_metrics()
file: tzrec/benchmark/benchmark.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 877
unit: def grid()
file: tzrec/ops/triton/triton_hstu_linear.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 878
unit: def backward()
file: tzrec/ops/triton/triton_hstu_linear.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 879
unit: def backward()
file: tzrec/ops/triton/triton_hstu_linear.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 880
unit: def backward()
file: tzrec/ops/triton/triton_hstu_linear.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 881
unit: def backward()
file: tzrec/ops/triton/triton_jagged_tensors.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 882
unit: def backward()
file: tzrec/ops/triton/triton_jagged_tensors.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 883
unit: def backward()
file: tzrec/ops/triton/triton_jagged_tensors.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 884
unit: def backward()
file: tzrec/ops/triton/triton_position.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 885
unit: def bwd_pre_hook()
file: tzrec/ops/triton/triton_position.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 886
unit: def backward()
file: tzrec/ops/triton/triton_position.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 887
unit: def backward()
file: tzrec/ops/triton/triton_hstu_attention.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 888
unit: def backward()
file: tzrec/ops/triton/triton_addmm.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 889
unit: def grid()
file: tzrec/ops/triton/triton_layer_norm.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 890
unit: def backward()
file: tzrec/ops/triton/triton_layer_norm.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 891
unit: def backward()
file: tzrec/ops/triton/triton_layer_norm.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 892
unit: def grid()
file: tzrec/ops/triton/triton_layer_norm.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 893
unit: def backward()
file: tzrec/ops/triton/triton_layer_norm.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 894
unit: def grid()
file: tzrec/ops/triton/triton_layer_norm.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 895
unit: def _label_mask()
file: tzrec/loss/jrc_loss.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 896
unit: def _diag_index()
file: tzrec/loss/jrc_loss.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1