id: 1
unit: def parse_args_neuronx()
file: optimum/commands/export/neuronx.py
start line: 0
end line: 0
size: 270 LOC
McCabe index: 1
number of parameters: 1

id: 2
unit: def preshard_hook()
file: optimum/neuron/models/inference/backend/modules/attention/gqa.py
start line: 0
end line: 0
size: 225 LOC
McCabe index: 11
number of parameters: 3

id: 3
unit: def __init__()
file: optimum/neuron/models/training/llama/modeling_llama.py
start line: 0
end line: 0
size: 124 LOC
McCabe index: 8
number of parameters: 4

id: 4
unit: def parse_args_neuron()
file: optimum/commands/export/neuron.py
start line: 0
end line: 0
size: 105 LOC
McCabe index: 1
number of parameters: 1

id: 5
unit: def main()
file: tools/auto_fill_neuron_cache.py
start line: 0
end line: 0
size: 82 LOC
McCabe index: 13
number of parameters: 0

id: 6
unit: def create_accelerator_and_postprocess()
file: optimum/neuron/trainers.py
start line: 0
end line: 0
size: 81 LOC
McCabe index: 9
number of parameters: 1

id: 7
unit: def forward()
file: optimum/exporters/neuron/model_wrappers.py
start line: 0
end line: 0
size: 80 LOC
McCabe index: 7
number of parameters: 3

id: 8
unit: def _kernel_enabled_mlp()
file: optimum/neuron/models/inference/llama/modeling_llama.py
start line: 0
end line: 0
size: 67 LOC
McCabe index: 9
number of parameters: 5

id: 9
unit: def training_function()
file: notebooks/text-classification/scripts/train.py
start line: 0
end line: 0
size: 61 LOC
McCabe index: 5
number of parameters: 1

id: 10
unit: def main()
file: optimum/exporters/neuron/__main__.py
start line: 0
end line: 0
size: 59 LOC
McCabe index: 5
number of parameters: 0

id: 11
unit: def main()
file: tools/create_examples_from_transformers.py
start line: 0
end line: 0
size: 59 LOC
McCabe index: 13
number of parameters: 0

id: 12
unit: def __post_init__()
file: optimum/neuron/training_args.py
start line: 0
end line: 0
size: 58 LOC
McCabe index: 16
number of parameters: 1

id: 13
unit: def parse_args()
file: notebooks/text-classification/scripts/train.py
start line: 0
end line: 0
size: 55 LOC
McCabe index: 2
number of parameters: 0

id: 14
unit: def _get_model_param_count()
file: optimum/neuron/utils/training_utils.py
start line: 0
end line: 0
size: 55 LOC
McCabe index: 15
number of parameters: 2

id: 15
unit: def forward()
file: optimum/neuron/models/training/loss_utils.py
start line: 0
end line: 0
size: 55 LOC
McCabe index: 8
number of parameters: 6

id: 16
unit: def perform_prefill()
file: optimum/neuron/models/inference/backend/modules/attention/attention_base.py
start line: 0
end line: 0
size: 55 LOC
McCabe index: 7
number of parameters: 7

id: 17
unit: def parse_args()
file: optimum/commands/neuron/cache.py
start line: 0
end line: 0
size: 53 LOC
McCabe index: 1
number of parameters: 1

id: 18
unit: def neff_cache()
file: optimum/neuron/models/inference/backend/cache.py
start line: 0
end line: 0
size: 52 LOC
McCabe index: 6
number of parameters: 1

id: 19
unit: def main()
file: benchmark/text-generation/performance/gen_barcharts.py
start line: 0
end line: 0
size: 51 LOC
McCabe index: 9
number of parameters: 0

id: 20
unit: def generate()
file: optimum/neuron/utils/input_generators.py
start line: 0
end line: 0
size: 50 LOC
McCabe index: 10
number of parameters: 5

id: 21
unit: def convert_mixtral_to_neuron_state_dict()
file: optimum/neuron/models/inference/mixtral/modeling_mixtral.py
start line: 0
end line: 0
size: 49 LOC
McCabe index: 5
number of parameters: 3

id: 22
unit: def get_node_results()
file: benchmark/text-generation-inference/performance/tgi_live_metrics.py
start line: 0
end line: 0
size: 47 LOC
McCabe index: 11
number of parameters: 1

id: 23
unit: def _maybe_log_save_evaluate()
file: optimum/neuron/trainers.py
start line: 0
end line: 0
size: 47 LOC
McCabe index: 12
number of parameters: 8

id: 24
unit: def __init__()
file: optimum/neuron/models/inference/llama/modeling_llama.py
start line: 0
end line: 0
size: 47 LOC
McCabe index: 3
number of parameters: 3

id: 25
unit: def get_relevant_parameter_names()
file: optimum/neuron/models/training/transformations_utils.py
start line: 0
end line: 0
size: 46 LOC
McCabe index: 14
number of parameters: 2

id: 26
unit: def __init__()
file: optimum/neuron/models/training/llama/modeling_llama.py
start line: 0
end line: 0
size: 45 LOC
McCabe index: 2
number of parameters: 3

id: 27
unit: def forward()
file: optimum/neuron/peft/tuners/lora/layer.py
start line: 0
end line: 0
size: 43 LOC
McCabe index: 11
number of parameters: 4

id: 28
unit: def new_fwd()
file: optimum/neuron/generation/utils.py
start line: 0
end line: 0
size: 42 LOC
McCabe index: 14
number of parameters: 2

id: 29
unit: def training_function()
file: notebooks/text-generation/scripts/sft_finetuning_qwen3.py
start line: 0
end line: 0
size: 41 LOC
McCabe index: 3
number of parameters: 2

id: 30
unit: def __call__()
file: optimum/neuron/pipelines/diffusers/pipeline_controlnet_sd_xl.py
start line: 0
end line: 0
size: 41 LOC
McCabe index: 1
number of parameters: 42

id: 31
unit: def run()
file: optimum/commands/env.py
start line: 0
end line: 0
size: 40 LOC
McCabe index: 4
number of parameters: 1

id: 32
unit: def _save_xla()
file: optimum/neuron/trainers.py
start line: 0
end line: 0
size: 40 LOC
McCabe index: 13
number of parameters: 2

id: 33
unit: def __init__()
file: optimum/neuron/models/inference/backend/config.py
start line: 0
end line: 0
size: 40 LOC
McCabe index: 2
number of parameters: 0

id: 34
unit: def _load_models_to_neuron()
file: optimum/neuron/modeling_diffusion.py
start line: 0
end line: 0
size: 37 LOC
McCabe index: 19
number of parameters: 3

id: 35
unit: def __init__()
file: optimum/neuron/peft/tuners/lora/layer.py
start line: 0
end line: 0
size: 37 LOC
McCabe index: 8
number of parameters: 4

id: 36
unit: def cumsum()
file: optimum/neuron/models/inference/backend/modules/generation/sampling.py
start line: 0
end line: 0
size: 37 LOC
McCabe index: 8
number of parameters: 3

id: 37
unit: def main()
file: benchmark/text-generation/performance/llama3.3-70b.py
start line: 0
end line: 0
size: 36 LOC
McCabe index: 4
number of parameters: 0

id: 38
unit: def forward()
file: optimum/exporters/neuron/model_wrappers.py
start line: 0
end line: 0
size: 36 LOC
McCabe index: 7
number of parameters: 2

id: 39
unit: def save_mutated_as_lora()
file: optimum/neuron/peft/peft_model.py
start line: 0
end line: 0
size: 35 LOC
McCabe index: 8
number of parameters: 4

id: 40
unit: def __init__()
file: optimum/neuron/accelerate/state.py
start line: 0
end line: 0
size: 35 LOC
McCabe index: 9
number of parameters: 3

id: 41
unit: def step()
file: optimum/neuron/accelerate/optimizer.py
start line: 0
end line: 0
size: 35 LOC
McCabe index: 12
number of parameters: 2

id: 42
unit: def _neuron_config_init()
file: optimum/neuron/modeling_traced.py
start line: 0
end line: 0
size: 34 LOC
McCabe index: 6
number of parameters: 2

id: 43
unit: def forward()
file: optimum/neuron/models/inference/backend/modules/decoder/decoder_wrapper.py
start line: 0
end line: 0
size: 34 LOC
McCabe index: 9
number of parameters: 6

id: 44
unit: def _load_weights_from_path()
file: optimum/neuron/models/inference/backend/pretrained_model.py
start line: 0
end line: 0
size: 33 LOC
McCabe index: 5
number of parameters: 2

id: 45
unit: def install_requirements()
file: optimum/neuron/utils/runner.py
start line: 0
end line: 0
size: 32 LOC
McCabe index: 7
number of parameters: 3

id: 46
unit: def _xla_gather()
file: optimum/neuron/accelerate/utils/operations.py
start line: 0
end line: 0
size: 32 LOC
McCabe index: 9
number of parameters: 2

id: 47
unit: def __init__()
file: optimum/neuron/modeling_diffusion.py
start line: 0
end line: 0
size: 31 LOC
McCabe index: 1
number of parameters: 0

id: 48
unit: def create_optimizer()
file: optimum/neuron/trainers.py
start line: 0
end line: 0
size: 31 LOC
McCabe index: 13
number of parameters: 1

id: 49
unit: def main_export()
file: optimum/exporters/neuron/__main__.py
start line: 0
end line: 0
size: 30 LOC
McCabe index: 1
number of parameters: 0

id: 50
unit: def __call__()
file: optimum/neuron/pipelines/diffusers/pipeline_controlnet.py
start line: 0
end line: 0
size: 30 LOC
McCabe index: 1
number of parameters: 0

id: 51
unit: def _init_kv_shape()
file: optimum/neuron/models/inference/backend/modules/kvcache/kv_cache_manager.py
start line: 0
end line: 0
size: 30 LOC
McCabe index: 4
number of parameters: 3

id: 52
unit: def to_dict()
file: optimum/neuron/configuration_utils.py
start line: 0
end line: 0
size: 30 LOC
McCabe index: 13
number of parameters: 1

id: 53
unit: def create_patched_save_pretrained()
file: optimum/neuron/accelerate/utils/misc.py
start line: 0
end line: 0
size: 30 LOC
McCabe index: 3
number of parameters: 2

id: 54
unit: def run()
file: benchmark/text-generation/performance/benchmark.py
start line: 0
end line: 0
size: 29 LOC
McCabe index: 3
number of parameters: 5

id: 55
unit: def main()
file: benchmark/text-generation-inference/performance/generate_csv.py
start line: 0
end line: 0
size: 29 LOC
McCabe index: 5
number of parameters: 0

id: 56
unit: def get_available_cores()
file: optimum/neuron/utils/system.py
start line: 0
end line: 0
size: 29 LOC
McCabe index: 12
number of parameters: 0

id: 57
unit: def _forward_with_pad()
file: optimum/neuron/models/inference/backend/modules/decoder/decoder_wrapper.py
start line: 0
end line: 0
size: 29 LOC
McCabe index: 9
number of parameters: 6

id: 58
unit: def forward()
file: optimum/neuron/models/inference/llama/modeling_llama.py
start line: 0
end line: 0
size: 29 LOC
McCabe index: 5
number of parameters: 3

id: 59
unit: def from_dict()
file: optimum/neuron/configuration_utils.py
start line: 0
end line: 0
size: 29 LOC
McCabe index: 9
number of parameters: 3

id: 60
unit: def _prepare_optimizer_for_zero_1()
file: optimum/neuron/accelerate/accelerator.py
start line: 0
end line: 0
size: 29 LOC
McCabe index: 3
number of parameters: 3

id: 61
unit: def run()
file: optimum/commands/neuron/cache.py
start line: 0
end line: 0
size: 28 LOC
McCabe index: 8
number of parameters: 1

id: 62
unit: def forward()
file: optimum/exporters/neuron/model_wrappers.py
start line: 0
end line: 0
size: 28 LOC
McCabe index: 3
number of parameters: 2

id: 63
unit: def patch_model_for_export()
file: optimum/exporters/neuron/model_configs.py
start line: 0
end line: 0
size: 28 LOC
McCabe index: 2
number of parameters: 4

id: 64
unit: def _kernel_qkv_forward()
file: optimum/neuron/models/inference/backend/modules/attention/gqa.py
start line: 0
end line: 0
size: 28 LOC
McCabe index: 6
number of parameters: 4

id: 65
unit: def main()
file: benchmark/text-generation/performance/llama3.1-8b.py
start line: 0
end line: 0
size: 27 LOC
McCabe index: 3
number of parameters: 0

id: 66
unit: def _export()
file: optimum/neuron/modeling_diffusion.py
start line: 0
end line: 0
size: 27 LOC
McCabe index: 1
number of parameters: 0

id: 67
unit: def convert_hf_to_neuron_state_dict()
file: optimum/neuron/models/inference/phi3/modeling_phi3.py
start line: 0
end line: 0
size: 27 LOC
McCabe index: 5
number of parameters: 3

id: 68
unit: def preshard_hook()
file: optimum/neuron/models/inference/backend/modules/attention/gqa.py
start line: 0
end line: 0
size: 27 LOC
McCabe index: 3
number of parameters: 3

id: 69
unit: def gather_for_metrics()
file: optimum/neuron/accelerate/accelerator.py
start line: 0
end line: 0
size: 27 LOC
McCabe index: 8
number of parameters: 2

id: 70
unit: def _save_checkpoint()
file: optimum/neuron/trainers.py
start line: 0
end line: 0
size: 26 LOC
McCabe index: 12
number of parameters: 3

id: 71
unit: def adapt_peft_config()
file: optimum/neuron/models/training/transformations_utils.py
start line: 0
end line: 0
size: 26 LOC
McCabe index: 12
number of parameters: 3

id: 72
unit: def _move_dict_args_to_device()
file: optimum/neuron/generation/utils.py
start line: 0
end line: 0
size: 25 LOC
McCabe index: 13
number of parameters: 3

id: 73
unit: def __init__()
file: optimum/neuron/trainers.py
start line: 0
end line: 0
size: 25 LOC
McCabe index: 12
number of parameters: 3

id: 74
unit: def download_example_script_from_github()
file: optimum/neuron/utils/runner.py
start line: 0
end line: 0
size: 25 LOC
McCabe index: 8
number of parameters: 3

id: 75
unit: def run()
file: optimum/neuron/utils/runner.py
start line: 0
end line: 0
size: 25 LOC
McCabe index: 1
number of parameters: 0

id: 76
unit: def _export()
file: optimum/neuron/modeling_traced.py
start line: 0
end line: 0
size: 25 LOC
McCabe index: 1
number of parameters: 0

id: 77
unit: def __init__()
file: optimum/neuron/models/training/qwen3/modeling_qwen3.py
start line: 0
end line: 0
size: 25 LOC
McCabe index: 4
number of parameters: 3

id: 78
unit: def backward()
file: optimum/neuron/models/training/loss_utils.py
start line: 0
end line: 0
size: 25 LOC
McCabe index: 4
number of parameters: 2

id: 79
unit: def _export()
file: optimum/neuron/modeling_seq2seq.py
start line: 0
end line: 0
size: 24 LOC
McCabe index: 1
number of parameters: 0

id: 80
unit: def __call__()
file: optimum/neuron/generation/logits_process.py
start line: 0
end line: 0
size: 24 LOC
McCabe index: 10
number of parameters: 2

id: 81
unit: def deprecate()
file: optimum/neuron/utils/deprecate_utils.py
start line: 0
end line: 0
size: 24 LOC
McCabe index: 7
number of parameters: 3

id: 82
unit: def create_venv()
file: optimum/neuron/utils/runner.py
start line: 0
end line: 0
size: 24 LOC
McCabe index: 3
number of parameters: 2

id: 83
unit: def _load_state_dict_into_model()
file: optimum/neuron/models/training/modeling_utils.py
start line: 0
end line: 0
size: 24 LOC
McCabe index: 13
number of parameters: 3

id: 84
unit: def create_nxdpp_model()
file: optimum/neuron/models/training/pipeline_utils.py
start line: 0
end line: 0
size: 24 LOC
McCabe index: 4
number of parameters: 1

id: 85
unit: def adapt_peft_config()
file: optimum/neuron/models/training/transformations_utils.py
start line: 0
end line: 0
size: 24 LOC
McCabe index: 12
number of parameters: 3

id: 86
unit: def compute_bias()
file: optimum/neuron/models/inference/t5/modeling_t5.py
start line: 0
end line: 0
size: 24 LOC
McCabe index: 3
number of parameters: 5

id: 87
unit: def create_model_wrappers()
file: optimum/neuron/models/inference/backend/modules/decoder/modeling_decoder.py
start line: 0
end line: 0
size: 24 LOC
McCabe index: 2
number of parameters: 4

id: 88
unit: def build_decoder_command()
file: tools/auto_fill_inference_cache.py
start line: 0
end line: 0
size: 24 LOC
McCabe index: 2
number of parameters: 6

id: 89
unit: def build_encoder_command()
file: tools/auto_fill_inference_cache.py
start line: 0
end line: 0
size: 24 LOC
McCabe index: 2
number of parameters: 7

id: 90
unit: def load_models_and_neuron_configs()
file: optimum/exporters/neuron/__main__.py
start line: 0
end line: 0
size: 23 LOC
McCabe index: 1
number of parameters: 0

id: 91
unit: def forward()
file: optimum/exporters/neuron/base.py
start line: 0
end line: 0
size: 23 LOC
McCabe index: 12
number of parameters: 2

id: 92
unit: def patch_model_for_export()
file: optimum/exporters/neuron/model_configs.py
start line: 0
end line: 0
size: 23 LOC
McCabe index: 3
number of parameters: 4

id: 93
unit: def get_hf_hub_cache_repos()
file: optimum/neuron/utils/cache_utils.py
start line: 0
end line: 0
size: 23 LOC
McCabe index: 13
number of parameters: 1

id: 94
unit: def __init__()
file: optimum/neuron/models/training/llama/modeling_llama.py
start line: 0
end line: 0
size: 23 LOC
McCabe index: 3
number of parameters: 3

id: 95
unit: def create_parameter_metadata()
file: optimum/neuron/models/training/transformations_utils.py
start line: 0
end line: 0
size: 23 LOC
McCabe index: 7
number of parameters: 1

id: 96
unit: def __init__()
file: optimum/neuron/models/inference/qwen3/modeling_qwen3.py
start line: 0
end line: 0
size: 23 LOC
McCabe index: 2
number of parameters: 3

id: 97
unit: def context_encoder_bk()
file: optimum/neuron/models/inference/backend/modules/autobucketing.py
start line: 0
end line: 0
size: 23 LOC
McCabe index: 10
number of parameters: 4

id: 98
unit: def __init__()
file: optimum/neuron/models/inference/qwen2/modeling_qwen2.py
start line: 0
end line: 0
size: 23 LOC
McCabe index: 2
number of parameters: 3

id: 99
unit: def __init__()
file: optimum/neuron/models/inference/llama/modeling_llama.py
start line: 0
end line: 0
size: 23 LOC
McCabe index: 2
number of parameters: 3

id: 100
unit: def get_dataset()
file: notebooks/text-generation/scripts/sft_finetuning_qwen3.py
start line: 0
end line: 0
size: 22 LOC
McCabe index: 2
number of parameters: 1

id: 101
unit: def pack_dataset()
file: notebooks/text-generation/scripts/utils/pack_dataset.py
start line: 0
end line: 0
size: 22 LOC
McCabe index: 12
number of parameters: 2

id: 102
unit: def reset_lora_parameters()
file: optimum/neuron/peft/tuners/lora/layer.py
start line: 0
end line: 0
size: 22 LOC
McCabe index: 10
number of parameters: 3

id: 103
unit: def has_write_access_to_repo()
file: optimum/neuron/utils/cache_utils.py
start line: 0
end line: 0
size: 22 LOC
McCabe index: 9
number of parameters: 1

id: 104
unit: def check_user_logged_in_and_cache_repo_is_set()
file: optimum/neuron/utils/runner.py
start line: 0
end line: 0
size: 22 LOC
McCabe index: 5
number of parameters: 1

id: 105
unit: def get_attention_scores_sd()
file: optimum/neuron/utils/optimization_utils.py
start line: 0
end line: 0
size: 22 LOC
McCabe index: 6
number of parameters: 4

id: 106
unit: def validate_sampling_params()
file: optimum/neuron/models/inference/backend/modules/generation/sampling.py
start line: 0
end line: 0
size: 22 LOC
McCabe index: 7
number of parameters: 2

id: 107
unit: def _top_k_masked()
file: optimum/neuron/models/inference/backend/modules/generation/sampling.py
start line: 0
end line: 0
size: 22 LOC
McCabe index: 3
number of parameters: 5

id: 108
unit: def __init__()
file: optimum/neuron/models/inference/mixtral/modeling_mixtral.py
start line: 0
end line: 0
size: 22 LOC
McCabe index: 2
number of parameters: 3

id: 109
unit: def apply_activation_checkpointing()
file: optimum/neuron/accelerate/utils/misc.py
start line: 0
end line: 0
size: 22 LOC
McCabe index: 10
number of parameters: 3

id: 110
unit: def infer_task_from_model_path()
file: tools/auto_fill_inference_cache.py
start line: 0
end line: 0
size: 22 LOC
McCabe index: 6
number of parameters: 1

id: 111
unit: def parse_args()
file: optimum/commands/neuron/subcommands.py
start line: 0
end line: 0
size: 21 LOC
McCabe index: 1
number of parameters: 1

id: 112
unit: def _create_dummy_input_generator_classes()
file: optimum/exporters/neuron/config.py
start line: 0
end line: 0
size: 21 LOC
McCabe index: 1
number of parameters: 2

id: 113
unit: def __post_init__()
file: optimum/neuron/models/training/config.py
start line: 0
end line: 0
size: 21 LOC
McCabe index: 7
number of parameters: 1

id: 114
unit: def suppress_logging()
file: optimum/neuron/models/training/pipeline_utils.py
start line: 0
end line: 0
size: 21 LOC
McCabe index: 4
number of parameters: 1

id: 115
unit: def prune_linear_layer()
file: optimum/neuron/models/inference/t5/modeling_t5.py
start line: 0
end line: 0
size: 21 LOC
McCabe index: 4
number of parameters: 3

id: 116
unit: def compute_for_token_gen()
file: optimum/neuron/models/inference/backend/modules/attention/attention_base.py
start line: 0
end line: 0
size: 21 LOC
McCabe index: 2
number of parameters: 8

id: 117
unit: def maybe_pad_interleaved()
file: optimum/neuron/models/inference/backend/modules/attention/gqa.py
start line: 0
end line: 0
size: 21 LOC
McCabe index: 9
number of parameters: 5

id: 118
unit: def __init__()
file: optimum/neuron/models/inference/qwen2/modeling_qwen2.py
start line: 0
end line: 0
size: 21 LOC
McCabe index: 1
number of parameters: 3

id: 119
unit: def __init__()
file: optimum/neuron/models/inference/llama/modeling_llama.py
start line: 0
end line: 0
size: 21 LOC
McCabe index: 1
number of parameters: 3

id: 120
unit: def __init__()
file: optimum/neuron/models/inference/granite/modeling_granite.py
start line: 0
end line: 0
size: 21 LOC
McCabe index: 1
number of parameters: 3

id: 121
unit: def step()
file: optimum/neuron/accelerate/scheduler.py
start line: 0
end line: 0
size: 21 LOC
McCabe index: 12
number of parameters: 3

id: 122
unit: def autocast()
file: optimum/neuron/accelerate/accelerator.py
start line: 0
end line: 0
size: 21 LOC
McCabe index: 5
number of parameters: 3

id: 123
unit: def main()
file: tools/decode_hlos.py
start line: 0
end line: 0
size: 21 LOC
McCabe index: 4
number of parameters: 0

id: 124
unit: def save_bar_chart()
file: benchmark/text-generation/performance/gen_barcharts.py
start line: 0
end line: 0
size: 20 LOC
McCabe index: 4
number of parameters: 6

id: 125
unit: def _load_lora_weights_to_pipeline()
file: optimum/exporters/neuron/utils.py
start line: 0
end line: 0
size: 20 LOC
McCabe index: 10
number of parameters: 2

id: 126
unit: def forward()
file: optimum/exporters/neuron/model_wrappers.py
start line: 0
end line: 0
size: 20 LOC
McCabe index: 2
number of parameters: 2

id: 127
unit: def generate_dummy_inputs()
file: optimum/exporters/neuron/model_configs.py
start line: 0
end line: 0
size: 20 LOC
McCabe index: 5
number of parameters: 3

id: 128
unit: def _from_pretrained()
file: optimum/neuron/modeling_diffusion.py
start line: 0
end line: 0
size: 20 LOC
McCabe index: 2
number of parameters: 0

id: 129
unit: def components()
file: optimum/neuron/modeling_diffusion.py
start line: 0
end line: 0
size: 20 LOC
McCabe index: 1
number of parameters: 1

id: 130
unit: def generate()
file: optimum/neuron/utils/input_generators.py
start line: 0
end line: 0
size: 20 LOC
McCabe index: 4
number of parameters: 5

id: 131
unit: def replace_class_in_inheritance_hierarchy()
file: optimum/neuron/utils/patching.py
start line: 0
end line: 0
size: 20 LOC
McCabe index: 7
number of parameters: 3

id: 132
unit: def create_context_encoding_wrapper()
file: optimum/neuron/models/inference/backend/modules/decoder/modeling_decoder.py
start line: 0
end line: 0
size: 20 LOC
McCabe index: 2
number of parameters: 4

id: 133
unit: def distributed_softmax()
file: optimum/neuron/models/inference/backend/modules/attention/utils.py
start line: 0
end line: 0
size: 20 LOC
McCabe index: 3
number of parameters: 2

id: 134
unit: def __init__()
file: optimum/neuron/models/inference/mixtral/modeling_mixtral.py
start line: 0
end line: 0
size: 20 LOC
McCabe index: 1
number of parameters: 4

id: 135
unit: def parse_args()
file: tools/create_examples_from_transformers.py
start line: 0
end line: 0
size: 20 LOC
McCabe index: 1
number of parameters: 0

id: 136
unit: def _initialize_past()
file: optimum/neuron/generation/utils.py
start line: 0
end line: 0
size: 19 LOC
McCabe index: 3
number of parameters: 2

id: 137
unit: def add_start_docstrings_to_model_forward()
file: optimum/neuron/utils/doc.py
start line: 0
end line: 0
size: 19 LOC
McCabe index: 8
number of parameters: 1

id: 138
unit: def validate_args()
file: optimum/neuron/hf_argparser.py
start line: 0
end line: 0
size: 19 LOC
McCabe index: 1
number of parameters: 2

id: 139
unit: def _fix_state_dict_keys_on_load()
file: optimum/neuron/models/training/modeling_utils.py
start line: 0
end line: 0
size: 19 LOC
McCabe index: 8
number of parameters: 2

id: 140
unit: def get_relevant_parameter_names()
file: optimum/neuron/models/training/transformations_utils.py
start line: 0
end line: 0
size: 19 LOC
McCabe index: 8
number of parameters: 2

id: 141
unit: def get_state_dict()
file: optimum/neuron/models/inference/backend/pretrained_model.py
start line: 0
end line: 0
size: 19 LOC
McCabe index: 6
number of parameters: 4

id: 142
unit: def __init__()
file: optimum/neuron/models/inference/backend/modules/decoder/modeling_decoder.py
start line: 0
end line: 0
size: 19 LOC
McCabe index: 4
number of parameters: 3

id: 143
unit: def create_speculation_wrapper()
file: optimum/neuron/models/inference/backend/modules/decoder/modeling_decoder.py
start line: 0
end line: 0
size: 19 LOC
McCabe index: 2
number of parameters: 4

id: 144
unit: def _multinomial_sample()
file: optimum/neuron/models/inference/backend/modules/generation/sampling.py
start line: 0
end line: 0
size: 19 LOC
McCabe index: 2
number of parameters: 6

id: 145
unit: def apply_scaling()
file: optimum/neuron/models/inference/backend/modules/attention/utils.py
start line: 0
end line: 0
size: 19 LOC
McCabe index: 5
number of parameters: 1

id: 146
unit: def __init__()
file: optimum/neuron/models/inference/backend/modules/attention/gqa.py
start line: 0
end line: 0
size: 19 LOC
McCabe index: 1
number of parameters: 0

id: 147
unit: def _split_fused_qkv()
file: optimum/neuron/models/inference/backend/modules/attention/gqa.py
start line: 0
end line: 0
size: 19 LOC
McCabe index: 5
number of parameters: 2

id: 148
unit: def to_json_string()
file: optimum/neuron/configuration_utils.py
start line: 0
end line: 0
size: 19 LOC
McCabe index: 8
number of parameters: 1

id: 149
unit: def main()
file: tools/prune_test_models.py
start line: 0
end line: 0
size: 19 LOC
McCabe index: 5
number of parameters: 0

id: 150
unit: def training_function()
file: notebooks/text-generation/scripts/run_clm.py
start line: 0
end line: 0
size: 18 LOC
McCabe index: 3
number of parameters: 2

id: 151
unit: def outputs()
file: optimum/exporters/neuron/model_configs.py
start line: 0
end line: 0
size: 18 LOC
McCabe index: 13
number of parameters: 1

id: 152
unit: def load_pipeline()
file: optimum/neuron/pipelines/transformers/base.py
start line: 0
end line: 0
size: 18 LOC
McCabe index: 1
number of parameters: 0

id: 153
unit: def forward()
file: optimum/neuron/peft/tuners/lora/layer.py
start line: 0
end line: 0
size: 18 LOC
McCabe index: 4
number of parameters: 4

id: 154
unit: def tokenize()
file: optimum/neuron/trainers.py
start line: 0
end line: 0
size: 18 LOC
McCabe index: 6
number of parameters: 1

id: 155
unit: def set_neuron_cache_path()
file: optimum/neuron/utils/cache_utils.py
start line: 0
end line: 0
size: 18 LOC
McCabe index: 5
number of parameters: 3

id: 156
unit: def patched_gradient_checkpointing_enable()
file: optimum/neuron/accelerate/utils/misc.py
start line: 0
end line: 0
size: 18 LOC
McCabe index: 6
number of parameters: 2

id: 157
unit: def unflatten_inputs()
file: optimum/exporters/neuron/base.py
start line: 0
end line: 0
size: 17 LOC
McCabe index: 7
number of parameters: 3

id: 158
unit: def beam_search()
file: optimum/neuron/generation/utils.py
start line: 0
end line: 0
size: 17 LOC
McCabe index: 1
number of parameters: 0

id: 159
unit: def is_private_repo()
file: optimum/neuron/utils/cache_utils.py
start line: 0
end line: 0
size: 17 LOC
McCabe index: 5
number of parameters: 1

id: 160
unit: def __post_init__()
file: optimum/neuron/utils/argument_utils.py
start line: 0
end line: 0
size: 17 LOC
McCabe index: 5
number of parameters: 1

id: 161
unit: def store_compilation_config()
file: optimum/neuron/utils/argument_utils.py
start line: 0
end line: 0
size: 17 LOC
McCabe index: 1
number of parameters: 0

id: 162
unit: def __init__()
file: optimum/neuron/models/training/llama/modeling_llama.py
start line: 0
end line: 0
size: 17 LOC
McCabe index: 2
number of parameters: 3

id: 163
unit: def convert_hf_to_neuron_state_dict()
file: optimum/neuron/models/inference/qwen3/modeling_qwen3.py
start line: 0
end line: 0
size: 17 LOC
McCabe index: 5
number of parameters: 3

id: 164
unit: def get_compiler_args()
file: optimum/neuron/models/inference/backend/modules/decoder/modeling_decoder.py
start line: 0
end line: 0
size: 17 LOC
McCabe index: 2
number of parameters: 2

id: 165
unit: def save_state_dict_safetensors()
file: optimum/neuron/models/inference/backend/modules/checkpoint.py
start line: 0
end line: 0
size: 17 LOC
McCabe index: 3
number of parameters: 3

id: 166
unit: def compute_for_flash_decoding()
file: optimum/neuron/models/inference/backend/modules/attention/attention_base.py
start line: 0
end line: 0
size: 17 LOC
McCabe index: 1
number of parameters: 7

id: 167
unit: def get_submodels_and_neuron_configs()
file: optimum/exporters/neuron/__main__.py
start line: 0
end line: 0
size: 16 LOC
McCabe index: 1
number of parameters: 0

id: 168
unit: def replace_weights()
file: optimum/neuron/modeling_diffusion.py
start line: 0
end line: 0
size: 16 LOC
McCabe index: 4
number of parameters: 4

id: 169
unit: def greedy_search()
file: optimum/neuron/generation/utils.py
start line: 0
end line: 0
size: 16 LOC
McCabe index: 1
number of parameters: 0

id: 170
unit: def check_compiler_compatibility()
file: optimum/neuron/utils/version_utils.py
start line: 0
end line: 0
size: 16 LOC
McCabe index: 5
number of parameters: 2

id: 171
unit: def get_tied_parameters_dict()
file: optimum/neuron/utils/model_utils.py
start line: 0
end line: 0
size: 16 LOC
McCabe index: 5
number of parameters: 2

id: 172
unit: def tie_parameters()
file: optimum/neuron/utils/model_utils.py
start line: 0
end line: 0
size: 16 LOC
McCabe index: 5
number of parameters: 4

id: 173
unit: def __init__()
file: optimum/neuron/models/training/granite/modeling_granite.py
start line: 0
end line: 0
size: 16 LOC
McCabe index: 3
number of parameters: 3

id: 174
unit: def _fix_state_dict_key_on_load()
file: optimum/neuron/models/training/modeling_utils.py
start line: 0
end line: 0
size: 16 LOC
McCabe index: 11
number of parameters: 1

id: 175
unit: def to_metadata()
file: optimum/neuron/models/training/transformations_utils.py
start line: 0
end line: 0
size: 16 LOC
McCabe index: 5
number of parameters: 2

id: 176
unit: def _forward()
file: optimum/neuron/models/inference/backend/modules/decoder/decoder_wrapper.py
start line: 0
end line: 0
size: 16 LOC
McCabe index: 7
number of parameters: 6

id: 177
unit: def pad_to_max_compiled_seq()
file: optimum/neuron/models/inference/backend/modules/decoder/decoder_wrapper.py
start line: 0
end line: 0
size: 16 LOC
McCabe index: 4
number of parameters: 2

id: 178
unit: def load_state_dict()
file: optimum/neuron/models/inference/backend/modules/checkpoint.py
start line: 0
end line: 0
size: 16 LOC
McCabe index: 5
number of parameters: 1

id: 179
unit: def pad_to_128_multiple()
file: optimum/neuron/models/inference/backend/modules/attention/utils.py
start line: 0
end line: 0
size: 16 LOC
McCabe index: 2
number of parameters: 2

id: 180
unit: def get_flash_attention_strategy()
file: optimum/neuron/models/inference/backend/modules/attention/attention_base.py
start line: 0
end line: 0
size: 16 LOC
McCabe index: 9
number of parameters: 2

id: 181
unit: def __init__()
file: optimum/neuron/models/inference/backend/modules/attention/gqa.py
start line: 0
end line: 0
size: 16 LOC
McCabe index: 1
number of parameters: 0

id: 182
unit: def apply_fp32_wrapper_to_vae_decoder()
file: optimum/exporters/neuron/utils.py
start line: 0
end line: 0
size: 15 LOC
McCabe index: 3
number of parameters: 1

id: 183
unit: def inputs()
file: optimum/exporters/neuron/model_configs.py
start line: 0
end line: 0
size: 15 LOC
McCabe index: 7
number of parameters: 1

id: 184
unit: def prepare_for_precompilation()
file: optimum/neuron/trainers.py
start line: 0
end line: 0
size: 15 LOC
McCabe index: 8
number of parameters: 2

id: 185
unit: def __init__()
file: optimum/neuron/trainers.py
start line: 0
end line: 0
size: 15 LOC
McCabe index: 1
number of parameters: 19

id: 186
unit: def is_main_worker_for_metrics()
file: optimum/neuron/utils/training_utils.py
start line: 0
end line: 0
size: 15 LOC
McCabe index: 3
number of parameters: 0

id: 187
unit: def map_torch_dtype()
file: optimum/neuron/utils/misc.py
start line: 0
end line: 0
size: 15 LOC
McCabe index: 3
number of parameters: 2

id: 188
unit: def neuron_scaled_dot_product_attention()
file: optimum/neuron/utils/optimization_utils.py
start line: 0
end line: 0
size: 15 LOC
McCabe index: 4
number of parameters: 6

id: 189
unit: def forward()
file: optimum/neuron/models/training/llama/modeling_llama.py
start line: 0
end line: 0
size: 15 LOC
McCabe index: 4
number of parameters: 3

id: 190
unit: def from_pretrained()
file: optimum/neuron/models/training/modeling_utils.py
start line: 0
end line: 0
size: 15 LOC
McCabe index: 1
number of parameters: 0

id: 191
unit: def _argmax_sample()
file: optimum/neuron/models/inference/backend/modules/generation/sampling.py
start line: 0
end line: 0
size: 15 LOC
McCabe index: 3
number of parameters: 4

id: 192
unit: def get_cache()
file: optimum/neuron/models/inference/backend/modules/kvcache/kv_cache_manager.py
start line: 0
end line: 0
size: 15 LOC
McCabe index: 5
number of parameters: 4

id: 193
unit: def __init__()
file: optimum/exporters/neuron/base.py
start line: 0
end line: 0
size: 14 LOC
McCabe index: 3
number of parameters: 0

id: 194
unit: def update_past()
file: optimum/exporters/neuron/model_wrappers.py
start line: 0
end line: 0
size: 14 LOC
McCabe index: 3
number of parameters: 2

id: 195
unit: def set_default_dp_mode()
file: optimum/neuron/modeling_diffusion.py
start line: 0
end line: 0
size: 14 LOC
McCabe index: 5
number of parameters: 1

id: 196
unit: def _from_pretrained()
file: optimum/neuron/modeling_seq2seq.py
start line: 0
end line: 0
size: 14 LOC
McCabe index: 1
number of parameters: 0

id: 197
unit: def _sanitize_parameters()
file: optimum/neuron/pipelines/transformers/sentence_transformers.py
start line: 0
end line: 0
size: 14 LOC
McCabe index: 5
number of parameters: 5

id: 198
unit: def pipeline()
file: optimum/neuron/pipelines/transformers/base.py
start line: 0
end line: 0
size: 14 LOC
McCabe index: 1
number of parameters: 0

id: 199
unit: def __init__()
file: optimum/neuron/peft/tuners/lora/layer.py
start line: 0
end line: 0
size: 14 LOC
McCabe index: 2
number of parameters: 9

id: 200
unit: def __init__()
file: optimum/neuron/peft/tuners/lora/layer.py
start line: 0
end line: 0
size: 14 LOC
McCabe index: 2
number of parameters: 9

id: 201
unit: def _expand_dict_for_generation()
file: optimum/neuron/generation/utils.py
start line: 0
end line: 0
size: 14 LOC
McCabe index: 7
number of parameters: 1

id: 202
unit: def __init__()
file: optimum/neuron/trainers.py
start line: 0
end line: 0
size: 14 LOC
McCabe index: 1
number of parameters: 17

id: 203
unit: def run_command_with_realtime_output()
file: optimum/neuron/utils/runner.py
start line: 0
end line: 0
size: 14 LOC
McCabe index: 5
number of parameters: 2

id: 204
unit: def generate()
file: optimum/neuron/utils/input_generators.py
start line: 0
end line: 0
size: 14 LOC
McCabe index: 4
number of parameters: 5

id: 205
unit: def _from_pretrained()
file: optimum/neuron/modeling_traced.py
start line: 0
end line: 0
size: 14 LOC
McCabe index: 1
number of parameters: 0

id: 206
unit: def save_pretrained()
file: optimum/neuron/models/training/modeling_utils.py
start line: 0
end line: 0
size: 14 LOC
McCabe index: 1
number of parameters: 0

id: 207
unit: def move_params_to_cpu()
file: optimum/neuron/models/training/pipeline_utils.py
start line: 0
end line: 0
size: 14 LOC
McCabe index: 5
number of parameters: 2

id: 208
unit: def to_original_peft_config()
file: optimum/neuron/models/training/transformations_utils.py
start line: 0
end line: 0
size: 14 LOC
McCabe index: 4
number of parameters: 3

id: 209
unit: def initialize_process_group()
file: optimum/neuron/models/inference/backend/modules/decoder/modeling_decoder.py
start line: 0
end line: 0
size: 14 LOC
McCabe index: 3
number of parameters: 2

id: 210
unit: def _create_context_attn_mask()
file: optimum/neuron/models/inference/backend/modules/decoder/modeling_decoder.py
start line: 0
end line: 0
size: 14 LOC
McCabe index: 5
number of parameters: 3

id: 211
unit: def export()
file: optimum/neuron/models/inference/backend/modules/decoder/modeling_decoder.py
start line: 0
end line: 0
size: 14 LOC
McCabe index: 1
number of parameters: 0

id: 212
unit: def mask_padded_logits()
file: optimum/neuron/models/inference/backend/modules/generation/sampling.py
start line: 0
end line: 0
size: 14 LOC
McCabe index: 5
number of parameters: 4

id: 213
unit: def prepare_sampling_params()
file: optimum/neuron/models/inference/backend/modules/generation/sampling.py
start line: 0
end line: 0
size: 14 LOC
McCabe index: 2
number of parameters: 4

id: 214
unit: def __init__()
file: optimum/neuron/models/inference/backend/modules/kvcache/kv_cache_manager.py
start line: 0
end line: 0
size: 14 LOC
McCabe index: 2
number of parameters: 4

id: 215
unit: def _native_qkv_forward()
file: optimum/neuron/models/inference/backend/modules/attention/gqa.py
start line: 0
end line: 0
size: 14 LOC
McCabe index: 3
number of parameters: 2

id: 216
unit: def convert_state_dict_to_fused_qkv()
file: optimum/neuron/models/inference/llama/modeling_llama.py
start line: 0
end line: 0
size: 14 LOC
McCabe index: 2
number of parameters: 2

id: 217
unit: def main()
file: tools/list_top_models.py
start line: 0
end line: 0
size: 14 LOC
McCabe index: 2
number of parameters: 0

id: 218
unit: def parse_args()
file: optimum/commands/neuron/cache.py
start line: 0
end line: 0
size: 13 LOC
McCabe index: 1
number of parameters: 1

id: 219
unit: def parse_args()
file: optimum/commands/neuron/cache.py
start line: 0
end line: 0
size: 13 LOC
McCabe index: 1
number of parameters: 1

id: 220
unit: def run()
file: optimum/commands/neuron/subcommands.py
start line: 0
end line: 0
size: 13 LOC
McCabe index: 3
number of parameters: 1

id: 221
unit: def export_models()
file: optimum/exporters/neuron/convert.py
start line: 0
end line: 0
size: 13 LOC
McCabe index: 1
number of parameters: 0

id: 222
unit: def parse_optlevel()
file: optimum/exporters/neuron/__main__.py
start line: 0
end line: 0
size: 13 LOC
McCabe index: 5
number of parameters: 1

id: 223
unit: def get_diffusion_models_for_export()
file: optimum/exporters/neuron/utils.py
start line: 0
end line: 0
size: 13 LOC
McCabe index: 1
number of parameters: 0

id: 224
unit: def get_input_args_for_task()
file: optimum/exporters/neuron/base.py
start line: 0
end line: 0
size: 13 LOC
McCabe index: 5
number of parameters: 2

id: 225
unit: def __init__()
file: optimum/neuron/modeling_seq2seq.py
start line: 0
end line: 0
size: 13 LOC
McCabe index: 1
number of parameters: 0

id: 226
unit: def train()
file: optimum/neuron/trainers.py
start line: 0
end line: 0
size: 13 LOC
McCabe index: 8
number of parameters: 3

id: 227
unit: def _create_requires_function()
file: optimum/neuron/utils/require_utils.py
start line: 0
end line: 0
size: 13 LOC
McCabe index: 2
number of parameters: 1

id: 228
unit: def __init__()
file: optimum/neuron/models/whisper/model.py
start line: 0
end line: 0
size: 13 LOC
McCabe index: 1
number of parameters: 0

id: 229
unit: def __init__()
file: optimum/neuron/models/training/llama/modeling_llama.py
start line: 0
end line: 0
size: 13 LOC
McCabe index: 3
number of parameters: 3

id: 230
unit: def fixed_cross_entropy()
file: optimum/neuron/models/training/loss_utils.py
start line: 0
end line: 0
size: 13 LOC
McCabe index: 5
number of parameters: 5

id: 231
unit: def to_original_peft_config()
file: optimum/neuron/models/training/transformations_utils.py
start line: 0
end line: 0
size: 13 LOC
McCabe index: 5
number of parameters: 3

id: 232
unit: def prune_heads()
file: optimum/neuron/models/inference/t5/modeling_t5.py
start line: 0
end line: 0
size: 13 LOC
McCabe index: 2
number of parameters: 2

id: 233
unit: def shard_checkpoint()
file: optimum/neuron/models/inference/backend/pretrained_model.py
start line: 0
end line: 0
size: 13 LOC
McCabe index: 2
number of parameters: 4

id: 234
unit: def _load_from_files()
file: optimum/neuron/models/inference/backend/modules/checkpoint.py
start line: 0
end line: 0
size: 13 LOC
McCabe index: 4
number of parameters: 3

id: 235
unit: def forward()
file: optimum/neuron/models/inference/backend/modules/attention/gqa.py
start line: 0
end line: 0
size: 13 LOC
McCabe index: 4
number of parameters: 3

id: 236
unit: def allreduce_sequence_parallel_gradients()
file: optimum/neuron/accelerate/optimizer.py
start line: 0
end line: 0
size: 13 LOC
McCabe index: 9
number of parameters: 1

id: 237
unit: def save_state_for_mp()
file: optimum/neuron/accelerate/accelerator.py
start line: 0
end line: 0
size: 13 LOC
McCabe index: 1
number of parameters: 3

id: 238
unit: def add_stable_diffusion_compiler_args()
file: optimum/exporters/neuron/convert.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 11
number of parameters: 2

id: 239
unit: def infer_task()
file: optimum/exporters/neuron/__main__.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 3
number of parameters: 1

id: 240
unit: def get_neuron_config_class()
file: optimum/exporters/neuron/__main__.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 2
number of parameters: 2

id: 241
unit: def normalize_sentence_transformers_input_shapes()
file: optimum/exporters/neuron/__main__.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 5
number of parameters: 1

id: 242
unit: def forward()
file: optimum/exporters/neuron/model_wrappers.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 2
number of parameters: 4

id: 243
unit: def outputs()
file: optimum/exporters/neuron/model_configs.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 2
number of parameters: 1

id: 244
unit: def load_model()
file: optimum/neuron/modeling_diffusion.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 1
number of parameters: 0

id: 245
unit: def _combine_encoder_decoder_config()
file: optimum/neuron/modeling_seq2seq.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 1
number of parameters: 3

id: 246
unit: def _get_add_time_ids_image_to_image()
file: optimum/neuron/pipelines/diffusers/pipeline_utils.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 1
number of parameters: 0

id: 247
unit: def __init__()
file: optimum/neuron/peft/tuners/lora/layer.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 1
number of parameters: 0

id: 248
unit: def _load_optimizer_and_scheduler()
file: optimum/neuron/trainers.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 3
number of parameters: 2

id: 249
unit: def __init__()
file: optimum/neuron/utils/input_generators.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 1
number of parameters: 0

id: 250
unit: def get_attention_scores_sdxl()
file: optimum/neuron/utils/optimization_utils.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 2
number of parameters: 4

id: 251
unit: def xser_load_on_cpu()
file: optimum/neuron/models/training/checkpointing.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 2
number of parameters: 1

id: 252
unit: def forward()
file: optimum/neuron/models/inference/t5/modeling_t5.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 1
number of parameters: 0

id: 253
unit: def save()
file: optimum/neuron/models/inference/backend/pretrained_model.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 3
number of parameters: 3

id: 254
unit: def _infer_attention_mask()
file: optimum/neuron/models/inference/backend/modules/decoder/modeling_decoder.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 2
number of parameters: 2

id: 255
unit: def _from_pretrained()
file: optimum/neuron/models/inference/backend/modules/decoder/modeling_decoder.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 1
number of parameters: 0

id: 256
unit: def get()
file: optimum/neuron/models/inference/backend/modules/decoder/decoder_wrapper.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 7
number of parameters: 3

id: 257
unit: def _get_top_k_num_stages()
file: optimum/neuron/models/inference/backend/modules/generation/sampling.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 6
number of parameters: 1

id: 258
unit: def apply_rotary_polar_compatible()
file: optimum/neuron/models/inference/backend/modules/attention/utils.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 1
number of parameters: 3

id: 259
unit: def manual_softmax()
file: optimum/neuron/models/inference/backend/modules/attention/utils.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 2
number of parameters: 3

id: 260
unit: def forward()
file: optimum/neuron/models/inference/backend/modules/attention/utils.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 2
number of parameters: 3

id: 261
unit: def replicate_kv()
file: optimum/neuron/models/inference/backend/modules/attention/gqa.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 2
number of parameters: 4

id: 262
unit: def __init__()
file: optimum/neuron/models/inference/granite/modeling_granite.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 1
number of parameters: 3

id: 263
unit: def _export()
file: optimum/neuron/modeling_decoder.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 1
number of parameters: 0

id: 264
unit: def __init__()
file: optimum/neuron/accelerate/state.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 1
number of parameters: 0

id: 265
unit: def compile_and_cache_model()
file: tools/auto_fill_inference_cache.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 1
number of parameters: 0

id: 266
unit: def export()
file: optimum/exporters/neuron/convert.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 1
number of parameters: 0

id: 267
unit: def _get_submodels_and_neuron_configs_for_encoder_decoder()
file: optimum/exporters/neuron/__main__.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 1
number of parameters: 0

id: 268
unit: def _create_dummy_input_generator_classes()
file: optimum/exporters/neuron/model_configs.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 2
number of parameters: 2

id: 269
unit: def _save_pretrained()
file: optimum/neuron/modeling_diffusion.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 1
number of parameters: 0

id: 270
unit: def __call__()
file: optimum/neuron/modeling_diffusion.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 5
number of parameters: 3

id: 271
unit: def forward()
file: optimum/neuron/modeling_diffusion.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 1
number of parameters: 0

id: 272
unit: def generate()
file: optimum/neuron/modeling_seq2seq.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 1
number of parameters: 0

id: 273
unit: def save_model()
file: optimum/neuron/trainers.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 6
number of parameters: 3

id: 274
unit: def get_hlos_from_run_log()
file: optimum/neuron/utils/neuron_parallel_compile.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 4
number of parameters: 1

id: 275
unit: def get_neuron_cache_path()
file: optimum/neuron/utils/cache_utils.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 3
number of parameters: 0

id: 276
unit: def get_model_name_or_path()
file: optimum/neuron/utils/cache_utils.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 4
number of parameters: 1

id: 277
unit: def decorator()
file: optimum/neuron/utils/patching.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 4
number of parameters: 1

id: 278
unit: def string_to_bool()
file: optimum/neuron/utils/misc.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 4
number of parameters: 2

id: 279
unit: def download_checkpoints_in_cache()
file: optimum/neuron/utils/misc.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 1
number of parameters: 0

id: 280
unit: def forward()
file: optimum/neuron/models/training/llama/modeling_llama.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 2
number of parameters: 2

id: 281
unit: def __init__()
file: optimum/neuron/models/training/llama/modeling_llama.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 1
number of parameters: 4

id: 282
unit: def _load_pretrained_model()
file: optimum/neuron/models/training/modeling_utils.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 1
number of parameters: 0

id: 283
unit: def create_local_fused_weight()
file: optimum/neuron/models/training/transformations_utils.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 3
number of parameters: 6

id: 284
unit: def to_dict()
file: optimum/neuron/models/inference/backend/config.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 8
number of parameters: 1

id: 285
unit: def __init__()
file: optimum/neuron/models/inference/backend/modules/attention/gqa.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 1
number of parameters: 0

id: 286
unit: def convert_hf_to_neuron_state_dict()
file: optimum/neuron/models/inference/llama/modeling_llama.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 4
number of parameters: 3

id: 287
unit: def clip_grad_norm_()
file: optimum/neuron/accelerate/accelerator.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 3
number of parameters: 5

id: 288
unit: def parse_args()
file: tools/auto_fill_neuron_cache.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 1
number of parameters: 0

id: 289
unit: def generate()
file: benchmark/text-generation/performance/benchmark.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 1
number of parameters: 3

id: 290
unit: def export_neuron()
file: optimum/exporters/neuron/convert.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 1
number of parameters: 0

id: 291
unit: def get_encoder_decoder_models_for_export()
file: optimum/exporters/neuron/utils.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 1
number of parameters: 0

id: 292
unit: def __init__()
file: optimum/exporters/neuron/model_wrappers.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 1
number of parameters: 0

id: 293
unit: def forward()
file: optimum/exporters/neuron/model_wrappers.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 2
number of parameters: 2

id: 294
unit: def inputs()
file: optimum/exporters/neuron/model_configs.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 1
number of parameters: 1

id: 295
unit: def _create_dummy_input_generator_classes()
file: optimum/exporters/neuron/model_configs.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 1
number of parameters: 2

id: 296
unit: def get_parallel_callable()
file: optimum/exporters/neuron/model_configs.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 1
number of parameters: 0

id: 297
unit: def do_classifier_free_guidance()
file: optimum/neuron/modeling_diffusion.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 5
number of parameters: 1

id: 298
unit: def forward()
file: optimum/neuron/modeling_diffusion.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 1
number of parameters: 0

id: 299
unit: def forward()
file: optimum/neuron/modeling_diffusion.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 1
number of parameters: 0

id: 300
unit: def forward()
file: optimum/neuron/modeling_seq2seq.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 1
number of parameters: 0

id: 301
unit: def _update_model_kwargs_for_xla_generation()
file: optimum/neuron/modeling_seq2seq.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 2
number of parameters: 0

id: 302
unit: def update_layer()
file: optimum/neuron/peft/tuners/lora/layer.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 1
number of parameters: 0

id: 303
unit: def update_layer()
file: optimum/neuron/peft/tuners/lora/layer.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 1
number of parameters: 0

id: 304
unit: def _update_model_kwargs_for_xla_generation()
file: optimum/neuron/generation/utils.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 1
number of parameters: 0

id: 305
unit: def generate()
file: optimum/neuron/generation/utils.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 1
number of parameters: 0

id: 306
unit: def compute_loss()
file: optimum/neuron/trainers.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 2
number of parameters: 5

id: 307
unit: def cross_entropy_loss()
file: optimum/neuron/trainers.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 2
number of parameters: 2

id: 308
unit: def is_accelerate_available()
file: optimum/neuron/utils/import_utils.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 3
number of parameters: 1

id: 309
unit: def is_trl_available()
file: optimum/neuron/utils/import_utils.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 4
number of parameters: 1

id: 310
unit: def is_peft_available()
file: optimum/neuron/utils/import_utils.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 3
number of parameters: 1

id: 311
unit: def add_shapes_to_config()
file: optimum/neuron/utils/argument_utils.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 4
number of parameters: 3

id: 312
unit: def _original_filename_to_safetensors_filename()
file: optimum/neuron/utils/misc.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 3
number of parameters: 1

id: 313
unit: def get_neuronxcc_version()
file: optimum/neuron/utils/version_utils.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 3
number of parameters: 0

id: 314
unit: def get_neuroncc_version()
file: optimum/neuron/utils/version_utils.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 3
number of parameters: 0

id: 315
unit: def get_torch_xla_version()
file: optimum/neuron/utils/version_utils.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 3
number of parameters: 0

id: 316
unit: def get_neuronx_distributed_version()
file: optimum/neuron/utils/version_utils.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 3
number of parameters: 0

id: 317
unit: def get_torch_version()
file: optimum/neuron/utils/version_utils.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 3
number of parameters: 0

id: 318
unit: def check_compiler_compatibility_for_stable_diffusion()
file: optimum/neuron/utils/version_utils.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 3
number of parameters: 0

id: 319
unit: def _dynamic_frequency_update()
file: optimum/neuron/models/training/llama/modeling_llama.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 5
number of parameters: 3

id: 320
unit: def eager_attention_forward()
file: optimum/neuron/models/training/llama/modeling_llama.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 1
number of parameters: 0

id: 321
unit: def _init_weights()
file: optimum/neuron/models/training/llama/modeling_llama.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 5
number of parameters: 2

id: 322
unit: def forward()
file: optimum/neuron/models/training/llama/modeling_llama.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 1
number of parameters: 0

id: 323
unit: def forward()
file: optimum/neuron/models/training/llama/modeling_llama.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 1
number of parameters: 0

id: 324
unit: def forward()
file: optimum/neuron/models/training/granite/modeling_granite.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 1
number of parameters: 0

id: 325
unit: def forward()
file: optimum/neuron/models/training/granite/modeling_granite.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 1
number of parameters: 0

id: 326
unit: def get_pipeline_parameters_for_current_stage()
file: optimum/neuron/models/training/pipeline_utils.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 4
number of parameters: 1

id: 327
unit: def from_metadata()
file: optimum/neuron/models/training/transformations_utils.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 2
number of parameters: 3

id: 328
unit: def parallelize()
file: optimum/neuron/models/inference/t5/modeling_t5.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 2
number of parameters: 1

id: 329
unit: def forward()
file: optimum/neuron/models/inference/backend/modules/decoder/modeling_decoder.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 1
number of parameters: 0

id: 330
unit: def forward()
file: optimum/neuron/models/inference/backend/modules/decoder/modeling_decoder.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 1
number of parameters: 0

id: 331
unit: def __init__()
file: optimum/neuron/models/inference/backend/modules/decoder/decoder_wrapper.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 1
number of parameters: 0

id: 332
unit: def load_module()
file: optimum/neuron/models/inference/backend/modules/decoder/decoder_wrapper.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 4
number of parameters: 1

id: 333
unit: def load_safetensors_sharded()
file: optimum/neuron/models/inference/backend/modules/checkpoint.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 1
number of parameters: 1

id: 334
unit: def load_pytorch_model_bin_sharded()
file: optimum/neuron/models/inference/backend/modules/checkpoint.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 1
number of parameters: 1

id: 335
unit: def _reshape_tiled_cache()
file: optimum/neuron/models/inference/backend/modules/kvcache/kv_cache_manager.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 1
number of parameters: 1

id: 336
unit: def update_cache()
file: optimum/neuron/models/inference/backend/modules/kvcache/kv_cache_manager.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 1
number of parameters: 0

id: 337
unit: def forward()
file: optimum/neuron/models/inference/backend/modules/attention/attention_base.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 1
number of parameters: 0

id: 338
unit: def replace_prefixes()
file: optimum/neuron/models/inference/backend/modules/attention/gqa.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 4
number of parameters: 4

id: 339
unit: def get_hash_module()
file: optimum/neuron/models/inference/backend/cache.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 4
number of parameters: 2

id: 340
unit: def _native_mlp()
file: optimum/neuron/models/inference/llama/modeling_llama.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 3
number of parameters: 3

id: 341
unit: def get_neuron_config()
file: optimum/neuron/modeling_decoder.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 1
number of parameters: 0

id: 342
unit: def _check_initialized()
file: optimum/neuron/accelerate/state.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 4
number of parameters: 4

id: 343
unit: def patch_accelerate_is_torch_xla_available()
file: optimum/neuron/accelerate/utils/misc.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 3
number of parameters: 0

id: 344
unit: def get_pip_pkgs_version()
file: optimum/commands/env.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 4
number of parameters: 2

id: 345
unit: def _list_entries()
file: optimum/commands/neuron/cache.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 3
number of parameters: 1

id: 346
unit: def export_neuronx()
file: optimum/exporters/neuron/convert.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 1
number of parameters: 0

id: 347
unit: def infer_compiler_kwargs()
file: optimum/exporters/neuron/__main__.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 5
number of parameters: 1

id: 348
unit: def _get_submodels_and_neuron_configs_for_stable_diffusion()
file: optimum/exporters/neuron/__main__.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 1
number of parameters: 0

id: 349
unit: def build_stable_diffusion_components_mandatory_shapes()
file: optimum/exporters/neuron/utils.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 1
number of parameters: 0

id: 350
unit: def load_controlnets()
file: optimum/exporters/neuron/utils.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 4
number of parameters: 2

id: 351
unit: def check_mandatory_input_shapes()
file: optimum/exporters/neuron/utils.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 4
number of parameters: 3

id: 352
unit: def __setattr__()
file: optimum/exporters/neuron/base.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 4
number of parameters: 3

id: 353
unit: def flatten_inputs()
file: optimum/exporters/neuron/base.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 4
number of parameters: 3

id: 354
unit: def forward()
file: optimum/exporters/neuron/model_wrappers.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 1
number of parameters: 0

id: 355
unit: def __init__()
file: optimum/exporters/neuron/model_wrappers.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 1
number of parameters: 0

id: 356
unit: def forward()
file: optimum/neuron/modeling_diffusion.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 4
number of parameters: 3

id: 357
unit: def forward()
file: optimum/neuron/modeling_diffusion.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 1
number of parameters: 0

id: 358
unit: def _save_config()
file: optimum/neuron/modeling_seq2seq.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 1
number of parameters: 2

id: 359
unit: def postprocess()
file: optimum/neuron/pipelines/transformers/sentence_transformers.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 8
number of parameters: 3

id: 360
unit: def _get_add_time_ids()
file: optimum/neuron/pipelines/diffusers/pipeline_utils.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 3
number of parameters: 3

id: 361
unit: def save_pretrained()
file: optimum/neuron/peft/peft_model.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 1
number of parameters: 0

id: 362
unit: def from_pretrained()
file: optimum/neuron/peft/peft_model.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 1
number of parameters: 0

id: 363
unit: def _update_attention()
file: optimum/neuron/generation/utils.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 2
number of parameters: 3

id: 364
unit: def __init__()
file: optimum/neuron/generation/token_selector.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 1
number of parameters: 0

id: 365
unit: def create()
file: optimum/neuron/generation/token_selector.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 1
number of parameters: 0

id: 366
unit: def _prepare_non_packed_dataloader()
file: optimum/neuron/trainers.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 1
number of parameters: 0

id: 367
unit: def is_torch_xla_available()
file: optimum/neuron/utils/import_utils.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 4
number of parameters: 0

id: 368
unit: def restore()
file: optimum/neuron/utils/patching.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 4
number of parameters: 1

id: 369
unit: def set_neuron_cc_optlevel()
file: optimum/neuron/utils/torch_xla_and_neuronx_initialization.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 2
number of parameters: 1

id: 370
unit: def check_neuron_cc_flags_for_model()
file: optimum/neuron/utils/torch_xla_and_neuronx_initialization.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 5
number of parameters: 1

id: 371
unit: def __init__()
file: optimum/neuron/modeling_traced.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 1
number of parameters: 0

id: 372
unit: def _prepare_4d_causal_attention_mask_with_cache_position()
file: optimum/neuron/models/training/qwen3/modeling_qwen3.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 1
number of parameters: 0

id: 373
unit: def _prepare_4d_causal_attention_mask_with_cache_position()
file: optimum/neuron/models/training/llama/modeling_llama.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 1
number of parameters: 0

id: 374
unit: def supports_pipeline_parallelism()
file: optimum/neuron/models/training/modeling_utils.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 4
number of parameters: 1

id: 375
unit: def get_tensor_model_parallel_attributes()
file: optimum/neuron/models/training/transformations_utils.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 2
number of parameters: 1

id: 376
unit: def __init__()
file: optimum/neuron/models/inference/t5/modeling_t5.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 1
number of parameters: 4

id: 377
unit: def _construct_output()
file: optimum/neuron/models/inference/backend/modules/decoder/modeling_decoder.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 2
number of parameters: 2

id: 378
unit: def _get_num_kv_heads_per_rank()
file: optimum/neuron/models/inference/backend/modules/kvcache/kv_cache_manager.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 1
number of parameters: 3

id: 379
unit: def _get_weight_from_state_dict()
file: optimum/neuron/models/inference/backend/modules/attention/utils.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 3
number of parameters: 3

id: 380
unit: def __init__()
file: optimum/neuron/models/inference/llama/modeling_llama.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 1
number of parameters: 0

id: 381
unit: def __init__()
file: optimum/neuron/models/inference/granite/modeling_granite.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 1
number of parameters: 3

id: 382
unit: def __init__()
file: optimum/neuron/models/inference/mixtral/modeling_mixtral.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 1
number of parameters: 3

id: 383
unit: def export()
file: optimum/neuron/modeling_decoder.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 1
number of parameters: 0

id: 384
unit: def _custom_save_state()
file: optimum/neuron/accelerate/accelerator.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 1
number of parameters: 0

id: 385
unit: def run_auto_fill_cache_for_model_name()
file: tools/auto_fill_neuron_cache.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 1
number of parameters: 0

id: 386
unit: def validate_models_outputs()
file: optimum/exporters/neuron/convert.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 1
number of parameters: 0

id: 387
unit: def improve_stable_diffusion_loading()
file: optimum/exporters/neuron/convert.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 6
number of parameters: 2

id: 388
unit: def maybe_export_from_neuron_model_class()
file: optimum/exporters/neuron/__main__.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 1
number of parameters: 0

id: 389
unit: def __init__()
file: optimum/exporters/neuron/model_wrappers.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 1
number of parameters: 0

id: 390
unit: def outputs()
file: optimum/exporters/neuron/model_configs.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 5
number of parameters: 1

id: 391
unit: def generate_io_aliases()
file: optimum/exporters/neuron/model_configs.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 4
number of parameters: 2

id: 392
unit: def generate_io_aliases()
file: optimum/exporters/neuron/model_configs.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 4
number of parameters: 2

id: 393
unit: def __init__()
file: optimum/neuron/modeling_diffusion.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 1
number of parameters: 0

id: 394
unit: def __init__()
file: optimum/neuron/modeling_seq2seq.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 1
number of parameters: 0

id: 395
unit: def forward()
file: optimum/neuron/modeling_seq2seq.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 1
number of parameters: 0

id: 396
unit: def load_adapter()
file: optimum/neuron/peft/peft_model.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 1
number of parameters: 0

id: 397
unit: def get_peft_model()
file: optimum/neuron/peft/mapping.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 1
number of parameters: 0

id: 398
unit: def _get_fwd_for_general_sampling()
file: optimum/neuron/generation/utils.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 1
number of parameters: 0

id: 399
unit: def _update_past()
file: optimum/neuron/generation/utils.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 3
number of parameters: 1

id: 400
unit: def synchronize_hub_cache()
file: optimum/neuron/trainers.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 4
number of parameters: 1

id: 401
unit: def _get_train_sampler()
file: optimum/neuron/trainers.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 5
number of parameters: 1

id: 402
unit: def get_neuron_major()
file: optimum/neuron/utils/system.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 3
number of parameters: 0

id: 403
unit: def is_trainium_test()
file: optimum/neuron/utils/testing_utils.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 2
number of parameters: 1

id: 404
unit: def is_inferentia_test()
file: optimum/neuron/utils/testing_utils.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 2
number of parameters: 1

id: 405
unit: def __init__()
file: optimum/neuron/utils/runner.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 1
number of parameters: 0

id: 406
unit: def lora_args()
file: optimum/neuron/utils/argument_utils.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 1
number of parameters: 1

id: 407
unit: def ip_adapter_args()
file: optimum/neuron/utils/argument_utils.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 1
number of parameters: 1

id: 408
unit: def distributed_friendly_cached_file()
file: optimum/neuron/utils/misc.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 3
number of parameters: 2

id: 409
unit: def saved_model_in_temporary_directory()
file: optimum/neuron/utils/model_utils.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 1
number of parameters: 1

id: 410
unit: def push_to_hub()
file: optimum/neuron/modeling_traced.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 1
number of parameters: 0

id: 411
unit: def auto_kv_size_multiplier()
file: optimum/neuron/models/training/config.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 3
number of parameters: 2

id: 412
unit: def forward()
file: optimum/neuron/models/training/llama/modeling_llama.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 2
number of parameters: 0

id: 413
unit: def forward()
file: optimum/neuron/models/training/granite/modeling_granite.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 2
number of parameters: 0

id: 414
unit: def _find_mismatched_keys()
file: optimum/neuron/models/training/modeling_utils.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 1
number of parameters: 0

id: 415
unit: def __init__()
file: optimum/neuron/models/inference/t5/modeling_t5.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 2
number of parameters: 2

id: 416
unit: def checkpoint_loader_fn()
file: optimum/neuron/models/inference/backend/pretrained_model.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 5
number of parameters: 4

id: 417
unit: def get_model_output()
file: optimum/neuron/models/inference/backend/modules/decoder/modeling_decoder.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 1
number of parameters: 0

id: 418
unit: def __init__()
file: optimum/neuron/models/inference/backend/modules/decoder/modeling_decoder.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 1
number of parameters: 0

id: 419
unit: def push_to_hub()
file: optimum/neuron/models/inference/backend/modules/decoder/modeling_decoder.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 1
number of parameters: 0

id: 420
unit: def get_model_instance()
file: optimum/neuron/models/inference/backend/modules/decoder/decoder_wrapper.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 1
number of parameters: 1

id: 421
unit: def prune_state_dict()
file: optimum/neuron/models/inference/backend/modules/checkpoint.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 5
number of parameters: 1

id: 422
unit: def _sample()
file: optimum/neuron/models/inference/backend/modules/generation/generation_utils.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 1
number of parameters: 0

id: 423
unit: def prepare_inputs_for_generation()
file: optimum/neuron/models/inference/backend/modules/generation/generation_utils.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 1
number of parameters: 0

id: 424
unit: def initialize_moe_module()
file: optimum/neuron/models/inference/backend/modules/moe.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 1
number of parameters: 0

id: 425
unit: def dynamic_update_slice()
file: optimum/neuron/models/inference/backend/modules/kvcache/utils.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 1
number of parameters: 3

id: 426
unit: def transpose_parallel_linear_layer()
file: optimum/neuron/models/inference/backend/modules/attention/utils.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 2
number of parameters: 1

id: 427
unit: def __init__()
file: optimum/neuron/models/inference/backend/modules/attention/attention_base.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 1
number of parameters: 0

id: 428
unit: def prep_qkv_tensors()
file: optimum/neuron/models/inference/backend/modules/attention/attention_base.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 1
number of parameters: 0

id: 429
unit: def maybe_pad_tail()
file: optimum/neuron/models/inference/backend/modules/attention/gqa.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 2
number of parameters: 4

id: 430
unit: def _get_neuron_config()
file: optimum/neuron/models/inference/qwen2/modeling_qwen2.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 1
number of parameters: 0

id: 431
unit: def _get_neuron_config()
file: optimum/neuron/models/inference/llama/modeling_llama.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 1
number of parameters: 0

id: 432
unit: def __init__()
file: optimum/neuron/models/inference/granite/modeling_granite.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 2
number of parameters: 3

id: 433
unit: def _get_neuron_config()
file: optimum/neuron/models/inference/mixtral/modeling_mixtral.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 1
number of parameters: 0

id: 434
unit: def _get_neuron_config()
file: optimum/neuron/modeling_decoder.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 1
number of parameters: 0

id: 435
unit: def __init__()
file: optimum/commands/export/neuron.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 0

id: 436
unit: def __init__()
file: optimum/commands/export/neuronx.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 0

id: 437
unit: def attention_wrapper()
file: optimum/exporters/neuron/utils.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 2
number of parameters: 6

id: 438
unit: def replace_stable_diffusion_submodels()
file: optimum/exporters/neuron/utils.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 3
number of parameters: 2

id: 439
unit: def patch_model_for_export()
file: optimum/exporters/neuron/base.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 0

id: 440
unit: def outputs()
file: optimum/exporters/neuron/model_configs.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 3
number of parameters: 1

id: 441
unit: def forward()
file: optimum/neuron/modeling_diffusion.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 0

id: 442
unit: def forward()
file: optimum/neuron/modeling_diffusion.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 0

id: 443
unit: def forward()
file: optimum/neuron/modeling_seq2seq.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 3

id: 444
unit: def prepare_inputs_for_generation()
file: optimum/neuron/modeling_seq2seq.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 0

id: 445
unit: def __init__()
file: optimum/neuron/peft/peft_model.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 0

id: 446
unit: def evaluation_loop()
file: optimum/neuron/trainers.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 0

id: 447
unit: def skip_first_batches()
file: optimum/neuron/utils/training_utils.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 2
number of parameters: 2

id: 448
unit: def get_model_param_count()
file: optimum/neuron/utils/training_utils.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 2
number of parameters: 3

id: 449
unit: def get_num_neuron_cores()
file: optimum/neuron/utils/cache_utils.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 2
number of parameters: 0

id: 450
unit: def list_files_in_neuron_cache()
file: optimum/neuron/utils/cache_utils.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 7
number of parameters: 3

id: 451
unit: def __init__()
file: optimum/neuron/utils/input_generators.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 0

id: 452
unit: def generate()
file: optimum/neuron/utils/input_generators.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 3
number of parameters: 5

id: 453
unit: def __init__()
file: optimum/neuron/utils/input_generators.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 0

id: 454
unit: def get_docstring_indentation_level()
file: optimum/neuron/utils/doc.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 2
number of parameters: 1

id: 455
unit: def init_process_group()
file: optimum/neuron/utils/torch_xla_and_neuronx_initialization.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 4
number of parameters: 0

id: 456
unit: def set_neuron_cc_flags_for_torch_amp()
file: optimum/neuron/utils/torch_xla_and_neuronx_initialization.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 2
number of parameters: 0

id: 457
unit: def world_size_as_dp_size()
file: optimum/neuron/training_args.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 1

id: 458
unit: def get_input_static_shapes()
file: optimum/neuron/modeling_traced.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 2
number of parameters: 2

id: 459
unit: def _validate_static_shape()
file: optimum/neuron/modeling_traced.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 4
number of parameters: 3

id: 460
unit: def _raise_if_invalid_padding()
file: optimum/neuron/modeling_traced.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 3
number of parameters: 6

id: 461
unit: def __init__()
file: optimum/neuron/models/training/granite/modeling_granite.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 2
number of parameters: 3

id: 462
unit: def _check_and_enable_flash_attn_2()
file: optimum/neuron/models/training/modeling_utils.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 0

id: 463
unit: def _autoset_attn_implementation()
file: optimum/neuron/models/training/modeling_utils.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 0

id: 464
unit: def __init__()
file: optimum/neuron/models/training/pipeline_utils.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 1

id: 465
unit: def _adapt_state_dict()
file: optimum/neuron/models/training/transformations_utils.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 0

id: 466
unit: def _lora_adapt_state_dict()
file: optimum/neuron/models/training/transformations_utils.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 0

id: 467
unit: def adapt_state_dict()
file: optimum/neuron/models/training/transformations_utils.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 0

id: 468
unit: def adapt_state_dict()
file: optimum/neuron/models/training/transformations_utils.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 0

id: 469
unit: def guess_peft_type()
file: optimum/neuron/models/training/transformations_utils.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 2
number of parameters: 3

id: 470
unit: def _adapt_state_dict()
file: optimum/neuron/models/training/transformations_utils.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 0

id: 471
unit: def _lora_adapt_state_dict()
file: optimum/neuron/models/training/transformations_utils.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 0

id: 472
unit: def create_gqa_query_or_output_projection_weight_from_full_weight()
file: optimum/neuron/models/training/transformations_utils.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 0

id: 473
unit: def guess_peft_type()
file: optimum/neuron/models/training/transformations_utils.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 2
number of parameters: 3

id: 474
unit: def _adapt_state_dict()
file: optimum/neuron/models/training/transformations_utils.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 0

id: 475
unit: def _lora_adapt_state_dict()
file: optimum/neuron/models/training/transformations_utils.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 0

id: 476
unit: def specialize_transformation_specs_for_model()
file: optimum/neuron/models/training/transformations_utils.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 4
number of parameters: 1

id: 477
unit: def __init__()
file: optimum/neuron/models/inference/t5/modeling_t5.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 2

id: 478
unit: def load_weights()
file: optimum/neuron/models/inference/backend/pretrained_model.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 0

id: 479
unit: def _create_chunked_prefill_attn_mask()
file: optimum/neuron/models/inference/backend/modules/decoder/modeling_decoder.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 0

id: 480
unit: def create_attn_mask()
file: optimum/neuron/models/inference/backend/modules/decoder/modeling_decoder.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 3
number of parameters: 5

id: 481
unit: def _slice_kv_cache()
file: optimum/neuron/models/inference/backend/modules/decoder/modeling_decoder.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 2
number of parameters: 3

id: 482
unit: def _setup_func_config()
file: optimum/neuron/models/inference/backend/modules/decoder/modeling_decoder.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 4
number of parameters: 4

id: 483
unit: def _get_model_outputs()
file: optimum/neuron/models/inference/backend/modules/decoder/modeling_decoder.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 0

id: 484
unit: def generate_buckets()
file: optimum/neuron/models/inference/backend/modules/autobucketing.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 5
number of parameters: 2

id: 485
unit: def _assisted_decoding()
file: optimum/neuron/models/inference/backend/modules/generation/generation_utils.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 0

id: 486
unit: def _fetch_cache()
file: optimum/neuron/models/inference/backend/modules/kvcache/kv_cache_manager.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 3
number of parameters: 3

id: 487
unit: def precompute_freqs_cis()
file: optimum/neuron/models/inference/backend/modules/attention/utils.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 2
number of parameters: 4

id: 488
unit: def set_weight()
file: optimum/neuron/models/inference/backend/modules/attention/gqa.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 0

id: 489
unit: def set_bias()
file: optimum/neuron/models/inference/backend/modules/attention/gqa.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 0

id: 490
unit: def __init__()
file: optimum/neuron/models/inference/llama/modeling_llama.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 0

id: 491
unit: def forward()
file: optimum/neuron/models/inference/llama/modeling_llama.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 0

id: 492
unit: def forward()
file: optimum/neuron/models/inference/granite/modeling_granite.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 0

id: 493
unit: def forward()
file: optimum/neuron/models/inference/mixtral/modeling_mixtral.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 0

id: 494
unit: def get_neuron_model_class()
file: optimum/neuron/models/auto_model.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 5
number of parameters: 4

id: 495
unit: def generate()
file: optimum/neuron/modeling_decoder.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 0

id: 496
unit: def __init__()
file: optimum/neuron/accelerate/accelerator.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 0

id: 497
unit: def backward()
file: optimum/neuron/accelerate/accelerator.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 3
number of parameters: 3

id: 498
unit: def forward()
file: optimum/neuron/modeling.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 0

id: 499
unit: def get_aws_neuronx_tools_version()
file: tools/auto_fill_inference_cache.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 3
number of parameters: 0

id: 500
unit: def remove_import()
file: tools/create_examples_from_transformers.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 2
number of parameters: 2

id: 501
unit: def remove_trainer_import()
file: tools/create_examples_from_transformers.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 2
number of parameters: 1

id: 502
unit: def generate_new_import_code()
file: tools/create_examples_from_transformers.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 3
number of parameters: 1

id: 503
unit: def get_testers_for_model_type()
file: tools/auto_fill_neuron_cache.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 5
number of parameters: 1

id: 504
unit: def print_apt_pkgs()
file: optimum/commands/env.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 2
number of parameters: 0

id: 505
unit: def validate_model_outputs()
file: optimum/exporters/neuron/convert.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 0

id: 506
unit: def customize_optional_outputs()
file: optimum/exporters/neuron/__main__.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 2
number of parameters: 1

id: 507
unit: def _validate_mandatory_axes()
file: optimum/exporters/neuron/base.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 3
number of parameters: 2

id: 508
unit: def __init__()
file: optimum/exporters/neuron/model_wrappers.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 4

id: 509
unit: def __init__()
file: optimum/exporters/neuron/model_wrappers.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 0

id: 510
unit: def __init__()
file: optimum/exporters/neuron/model_wrappers.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 0

id: 511
unit: def outputs()
file: optimum/exporters/neuron/model_configs.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 2
number of parameters: 1

id: 512
unit: def inputs()
file: optimum/exporters/neuron/model_configs.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 2
number of parameters: 1

id: 513
unit: def generate_dummy_inputs()
file: optimum/exporters/neuron/model_configs.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 2
number of parameters: 3

id: 514
unit: def generate_dummy_inputs()
file: optimum/exporters/neuron/model_configs.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 2

id: 515
unit: def __init__()
file: optimum/neuron/modeling_diffusion.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 0

id: 516
unit: def forward()
file: optimum/neuron/modeling_diffusion.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 0

id: 517
unit: def __init__()
file: optimum/neuron/modeling_diffusion.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 0

id: 518
unit: def __init__()
file: optimum/neuron/modeling_diffusion.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 0

id: 519
unit: def __init__()
file: optimum/neuron/modeling_diffusion.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 0

id: 520
unit: def __init__()
file: optimum/neuron/modeling_diffusion.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 0

id: 521
unit: def __init__()
file: optimum/neuron/modeling_diffusion.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 0

id: 522
unit: def __init__()
file: optimum/neuron/modeling_diffusion.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 0

id: 523
unit: def __init__()
file: optimum/neuron/modeling_diffusion.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 0

id: 524
unit: def __init__()
file: optimum/neuron/modeling_seq2seq.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 0

id: 525
unit: def __init__()
file: optimum/neuron/modeling_seq2seq.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 0

id: 526
unit: def is_sentence_transformer_model()
file: optimum/neuron/pipelines/transformers/sentence_transformers.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 2
number of parameters: 3

id: 527
unit: def __init__()
file: optimum/neuron/modeling_base.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 2
number of parameters: 3

id: 528
unit: def _initialize_attention()
file: optimum/neuron/generation/utils.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 0

id: 529
unit: def select()
file: optimum/neuron/generation/token_selector.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 2
number of parameters: 3

id: 530
unit: def _sample()
file: optimum/neuron/generation/token_selector.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 2
number of parameters: 2

id: 531
unit: def autocast_smart_context_manager()
file: optimum/neuron/trainers.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 2

id: 532
unit: def prediction_step()
file: optimum/neuron/trainers.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 0

id: 533
unit: def train()
file: optimum/neuron/trainers.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 2
number of parameters: 0

id: 534
unit: def is_custom_modeling_model()
file: optimum/neuron/utils/training_utils.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 2
number of parameters: 1

id: 535
unit: def compute_max_train_samples()
file: optimum/neuron/utils/runner.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 0

id: 536
unit: def __init__()
file: optimum/neuron/utils/input_generators.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 0

id: 537
unit: def __init__()
file: optimum/neuron/utils/input_generators.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 0

id: 538
unit: def validate_arg()
file: optimum/neuron/utils/argument_utils.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 0

id: 539
unit: def patch()
file: optimum/neuron/utils/patching.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 3
number of parameters: 1

id: 540
unit: def patch_everywhere()
file: optimum/neuron/utils/patching.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 5
number of parameters: 3

id: 541
unit: def is_main_worker()
file: optimum/neuron/utils/misc.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 4
number of parameters: 1

id: 542
unit: def set_common_flags()
file: optimum/neuron/utils/torch_xla_and_neuronx_initialization.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 2
number of parameters: 0

id: 543
unit: def world_size_should_behave_as_dp_size()
file: optimum/neuron/training_args.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 2
number of parameters: 2

id: 544
unit: def __init__()
file: optimum/neuron/models/whisper/model.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 0

id: 545
unit: def __init__()
file: optimum/neuron/models/whisper/model.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 0

id: 546
unit: def _update_model_kwargs_for_generation()
file: optimum/neuron/models/whisper/model.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 0

id: 547
unit: def forward()
file: optimum/neuron/models/whisper/model.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 0

id: 548
unit: def forward()
file: optimum/neuron/models/bert/model.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 0

id: 549
unit: def forward()
file: optimum/neuron/models/bert/model.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 0

id: 550
unit: def forward()
file: optimum/neuron/models/bert/model.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 0

id: 551
unit: def forward()
file: optimum/neuron/models/bert/model.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 0

id: 552
unit: def forward()
file: optimum/neuron/models/bert/model.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 0

id: 553
unit: def forward()
file: optimum/neuron/models/bert/model.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 0

id: 554
unit: def forward()
file: optimum/neuron/models/training/qwen3/modeling_qwen3.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 0

id: 555
unit: def forward()
file: optimum/neuron/models/training/llama/modeling_llama.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 2

id: 556
unit: def apply_rotary_pos_emb()
file: optimum/neuron/models/training/llama/modeling_llama.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 6

id: 557
unit: def repeat_kv()
file: optimum/neuron/models/training/llama/modeling_llama.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 2
number of parameters: 2

id: 558
unit: def forward()
file: optimum/neuron/models/training/llama/modeling_llama.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 0

id: 559
unit: def apply_rotary_pos_emb()
file: optimum/neuron/models/training/granite/modeling_granite.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 6

id: 560
unit: def repeat_kv()
file: optimum/neuron/models/training/granite/modeling_granite.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 2
number of parameters: 2

id: 561
unit: def _get_resized_embeddings()
file: optimum/neuron/models/training/modeling_utils.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 0

id: 562
unit: def _get_resized_lm_head()
file: optimum/neuron/models/training/modeling_utils.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 0

id: 563
unit: def _get_resized_parallel_embeddings()
file: optimum/neuron/models/training/modeling_utils.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 0

id: 564
unit: def _get_resized_parallel_lm_head()
file: optimum/neuron/models/training/modeling_utils.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 0

id: 565
unit: def create_query_or_output_projection_local_weight_from_regular_weight()
file: optimum/neuron/models/training/transformations_utils.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 0

id: 566
unit: def adapt_state_dict()
file: optimum/neuron/models/training/transformations_utils.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 0

id: 567
unit: def __init__()
file: optimum/neuron/models/inference/t5/modeling_t5.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 2

id: 568
unit: def get_builder()
file: optimum/neuron/models/inference/backend/pretrained_model.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 0

id: 569
unit: def __init__()
file: optimum/neuron/models/inference/backend/pretrained_model.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 0

id: 570
unit: def set_random_seed()
file: optimum/neuron/models/inference/backend/utils/random.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 2
number of parameters: 1

id: 571
unit: def _create_spec_attn_mask()
file: optimum/neuron/models/inference/backend/modules/decoder/modeling_decoder.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 2

id: 572
unit: def _torch_load()
file: optimum/neuron/models/inference/backend/modules/checkpoint.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 2
number of parameters: 1

id: 573
unit: def calculate_num_cores_per_group()
file: optimum/neuron/models/inference/backend/modules/flashdecode/utils.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 3

id: 574
unit: def rand_like()
file: optimum/neuron/models/inference/backend/modules/generation/sampling.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 1

id: 575
unit: def prepare_tensor()
file: optimum/neuron/models/inference/backend/modules/generation/sampling.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 3
number of parameters: 3

id: 576
unit: def _top_p()
file: optimum/neuron/models/inference/backend/modules/generation/sampling.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 5

id: 577
unit: def _multinomial()
file: optimum/neuron/models/inference/backend/modules/generation/sampling.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 4

id: 578
unit: def forward()
file: optimum/neuron/models/inference/backend/modules/generation/sampling.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 2
number of parameters: 5

id: 579
unit: def generate()
file: optimum/neuron/models/inference/backend/modules/generation/generation_utils.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 0

id: 580
unit: def prepare_inputs_for_prefill()
file: optimum/neuron/models/inference/backend/modules/generation/generation_utils.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 0

id: 581
unit: def prepare_inputs_for_decode()
file: optimum/neuron/models/inference/backend/modules/generation/generation_utils.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 0

id: 582
unit: def fill_prefix()
file: optimum/neuron/models/inference/backend/modules/kvcache/utils.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 2

id: 583
unit: def repeat_kv()
file: optimum/neuron/models/inference/backend/modules/attention/utils.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 2
number of parameters: 2

id: 584
unit: def apply_rotary_pos_emb()
file: optimum/neuron/models/inference/backend/modules/attention/utils.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 6

id: 585
unit: def __init__()
file: optimum/neuron/models/inference/backend/modules/attention/utils.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 4

id: 586
unit: def forward()
file: optimum/neuron/models/inference/llama/modeling_llama.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 2
number of parameters: 4

id: 587
unit: def get_compiler_args()
file: optimum/neuron/models/inference/mixtral/modeling_mixtral.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 2
number of parameters: 2

id: 588
unit: def forward()
file: optimum/neuron/models/clip/model.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 0

id: 589
unit: def register_neuron_config()
file: optimum/neuron/configuration_utils.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 1

id: 590
unit: def __eq__()
file: optimum/neuron/configuration_utils.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 2
number of parameters: 2

id: 591
unit: def save_pretrained()
file: optimum/neuron/configuration_utils.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 0

id: 592
unit: def from_pretrained()
file: optimum/neuron/configuration_utils.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 0

id: 593
unit: def __getstate__()
file: optimum/neuron/accelerate/optimizer.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 1

id: 594
unit: def torch_xla_safe_save_file()
file: optimum/neuron/accelerate/utils/misc.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 0

id: 595
unit: def _prepare_data_loader_for_distributed()
file: optimum/neuron/accelerate/accelerator.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 0

id: 596
unit: def forward()
file: optimum/neuron/modeling.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 0

id: 597
unit: def forward()
file: optimum/neuron/modeling.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 0

id: 598
unit: def forward()
file: optimum/neuron/modeling.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 0

id: 599
unit: def forward()
file: optimum/neuron/modeling.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 0

id: 600
unit: def forward()
file: optimum/neuron/modeling.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 0

id: 601
unit: def forward()
file: optimum/neuron/modeling.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 0

id: 602
unit: def keep_only_examples_with_trainer_and_requirements_predicate()
file: tools/create_examples_from_transformers.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 4
number of parameters: 1

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

id: 604
unit: def main()
file: notebooks/text-generation/scripts/sft_finetuning_qwen3.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 0

id: 605
unit: def main()
file: notebooks/text-generation/scripts/run_clm.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 0

id: 606
unit: def run()
file: optimum/commands/neuron/cache.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 2
number of parameters: 1

id: 607
unit: def parse_args()
file: optimum/commands/neuron/cache.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 1

id: 608
unit: def get_submodels_for_export_diffusion()
file: optimum/exporters/neuron/utils.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 0

id: 609
unit: def __getattr__()
file: optimum/exporters/neuron/base.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 3
number of parameters: 2

id: 610
unit: def __init__()
file: optimum/exporters/neuron/model_wrappers.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 4

id: 611
unit: def __init__()
file: optimum/exporters/neuron/model_wrappers.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 4

id: 612
unit: def reorder_cache()
file: optimum/exporters/neuron/model_wrappers.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 2
number of parameters: 3

id: 613
unit: def __init__()
file: optimum/exporters/neuron/model_wrappers.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 4

id: 614
unit: def __init__()
file: optimum/exporters/neuron/model_wrappers.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 4

id: 615
unit: def forward()
file: optimum/exporters/neuron/model_wrappers.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 0

id: 616
unit: def forward()
file: optimum/exporters/neuron/model_wrappers.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 0

id: 617
unit: def inputs()
file: optimum/exporters/neuron/model_configs.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 2
number of parameters: 1

id: 618
unit: def outputs()
file: optimum/exporters/neuron/model_configs.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 2
number of parameters: 1

id: 619
unit: def inputs()
file: optimum/exporters/neuron/model_configs.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 2
number of parameters: 1

id: 620
unit: def outputs()
file: optimum/exporters/neuron/model_configs.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 2
number of parameters: 1

id: 621
unit: def outputs()
file: optimum/exporters/neuron/model_configs.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 2
number of parameters: 1

id: 622
unit: def inputs()
file: optimum/exporters/neuron/model_configs.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 2
number of parameters: 1

id: 623
unit: def outputs()
file: optimum/exporters/neuron/model_configs.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 2
number of parameters: 1

id: 624
unit: def outputs()
file: optimum/exporters/neuron/model_configs.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 2
number of parameters: 1

id: 625
unit: def with_ip_adapter()
file: optimum/exporters/neuron/model_configs.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 2
number of parameters: 2

id: 626
unit: def patch_model_for_export()
file: optimum/exporters/neuron/model_configs.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 0

id: 627
unit: def parameters()
file: optimum/neuron/modeling_diffusion.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 1

id: 628
unit: def has_valid_embedding_base_layer()
file: optimum/neuron/peft/utils/save_and_load.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 2
number of parameters: 1

id: 629
unit: def to()
file: optimum/neuron/modeling_base.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 3
number of parameters: 3

id: 630
unit: def generate()
file: optimum/neuron/generation/utils.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 0

id: 631
unit: def _expand_inputs_for_generation()
file: optimum/neuron/generation/utils.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 0

id: 632
unit: def evaluate()
file: optimum/neuron/trainers.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 0

id: 633
unit: def is_topology_supported()
file: optimum/neuron/utils/training_utils.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 2
number of parameters: 0

id: 634
unit: def generate()
file: optimum/neuron/utils/input_generators.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 3
number of parameters: 5

id: 635
unit: def generate()
file: optimum/neuron/utils/input_generators.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 2
number of parameters: 5

id: 636
unit: def __init__()
file: optimum/neuron/utils/argument_utils.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 6
number of parameters: 2

id: 637
unit: def __init__()
file: optimum/neuron/utils/argument_utils.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 3
number of parameters: 2

id: 638
unit: def add_start_docstrings()
file: optimum/neuron/utils/doc.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 2
number of parameters: 1

id: 639
unit: def args_and_kwargs_to_kwargs_only()
file: optimum/neuron/utils/misc.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 0

id: 640
unit: def convert_checkpoint_to_safetensors()
file: optimum/neuron/utils/misc.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 0

id: 641
unit: def dp_size()
file: optimum/neuron/training_args.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 2
number of parameters: 1

id: 642
unit: def _attributes_init()
file: optimum/neuron/modeling_traced.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 0

id: 643
unit: def remove_padding()
file: optimum/neuron/modeling_traced.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 0

id: 644
unit: def parse_args_into_dataclasses()
file: optimum/neuron/hf_argparser.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 2
number of parameters: 3

id: 645
unit: def parse_dict()
file: optimum/neuron/hf_argparser.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 2
number of parameters: 3

id: 646
unit: def parse_json_file()
file: optimum/neuron/hf_argparser.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 2
number of parameters: 3

id: 647
unit: def parse_yaml_file()
file: optimum/neuron/hf_argparser.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 2
number of parameters: 3

id: 648
unit: def forward()
file: optimum/neuron/models/whisper/model.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 0

id: 649
unit: def forward()
file: optimum/neuron/models/whisper/model.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 0

id: 650
unit: def __init__()
file: optimum/neuron/models/training/llama/modeling_llama.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 4

id: 651
unit: def _update_causal_mask()
file: optimum/neuron/models/training/llama/modeling_llama.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 0

id: 652
unit: def resize_token_embeddings()
file: optimum/neuron/models/training/modeling_utils.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 0

id: 653
unit: def consolidate_tensor_parallel_checkpoints()
file: optimum/neuron/models/training/checkpointing.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 0

id: 654
unit: def create_local_weight_with_padding()
file: optimum/neuron/models/training/transformations_utils.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 0

id: 655
unit: def _to_original_weights()
file: optimum/neuron/models/training/transformations_utils.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 0

id: 656
unit: def _lora_to_original_weights()
file: optimum/neuron/models/training/transformations_utils.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 0

id: 657
unit: def to_original_weights()
file: optimum/neuron/models/training/transformations_utils.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 0

id: 658
unit: def __post_init__()
file: optimum/neuron/models/training/transformations_utils.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 3
number of parameters: 1

id: 659
unit: def _to_original_weights()
file: optimum/neuron/models/training/transformations_utils.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 0

id: 660
unit: def _lora_to_original_weights()
file: optimum/neuron/models/training/transformations_utils.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 0

id: 661
unit: def _to_original_weights()
file: optimum/neuron/models/training/transformations_utils.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 0

id: 662
unit: def _lora_to_original_weights()
file: optimum/neuron/models/training/transformations_utils.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 0

id: 663
unit: def get_adapter_name()
file: optimum/neuron/models/training/transformations_utils.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 2
number of parameters: 1

id: 664
unit: def to_original_weights()
file: optimum/neuron/models/training/transformations_utils.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 0

id: 665
unit: def __init__()
file: optimum/neuron/models/inference/t5/modeling_t5.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 0

id: 666
unit: def __init__()
file: optimum/neuron/models/inference/t5/modeling_t5.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 3

id: 667
unit: def __init__()
file: optimum/neuron/models/inference/qwen3/modeling_qwen3.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 3

id: 668
unit: def get_init_world_size()
file: optimum/neuron/models/inference/backend/utils/distributed.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 4
number of parameters: 0

id: 669
unit: def get_init_rank()
file: optimum/neuron/models/inference/backend/utils/distributed.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 4
number of parameters: 0

id: 670
unit: def __init__()
file: optimum/neuron/models/inference/backend/modules/generation/generation_utils.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 4

id: 671
unit: def _update_model_kwargs_for_generation()
file: optimum/neuron/models/inference/backend/modules/generation/generation_utils.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 0

id: 672
unit: def _slice_kv_cacheline()
file: optimum/neuron/models/inference/backend/modules/kvcache/kv_cache_manager.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 2
number of parameters: 3

id: 673
unit: def _get_hidden_dim_per_head()
file: optimum/neuron/models/inference/backend/modules/kvcache/kv_cache_manager.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 2

id: 674
unit: def forward()
file: optimum/neuron/models/inference/backend/modules/custom_calls.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 2

id: 675
unit: def _set_weight_to_state_dict()
file: optimum/neuron/models/inference/backend/modules/attention/utils.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 2
number of parameters: 4

id: 676
unit: def move_heads_front()
file: optimum/neuron/models/inference/backend/modules/attention/utils.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 2
number of parameters: 6

id: 677
unit: def create_block_diagonal_attn_mask()
file: optimum/neuron/models/inference/backend/modules/attention/utils.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 0

id: 678
unit: def scaled_qk()
file: optimum/neuron/models/inference/backend/modules/attention/attention_base.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 2
number of parameters: 4

id: 679
unit: def register_neuron_model()
file: optimum/neuron/models/auto_model.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 4

id: 680
unit: def get_neuron_causal_lm_model_class()
file: optimum/neuron/modeling_decoder.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 2
number of parameters: 1

id: 681
unit: def _from_pretrained()
file: optimum/neuron/modeling_decoder.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 0

id: 682
unit: def __init__()
file: optimum/neuron/accelerate/optimizer.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 0

id: 683
unit: def download_examples_from_transformers()
file: tools/create_examples_from_transformers.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 0

id: 684
unit: def open_and_append_to_file()
file: tools/auto_fill_neuron_cache.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 2

id: 685
unit: def get_input_shapes()
file: optimum/exporters/neuron/__main__.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 2
number of parameters: 2

id: 686
unit: def infer_stable_diffusion_shapes_from_diffusers()
file: optimum/exporters/neuron/__main__.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 0

id: 687
unit: def _create_dummy_input_generator_classes()
file: optimum/exporters/neuron/base.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 3
number of parameters: 2

id: 688
unit: def __init__()
file: optimum/exporters/neuron/base.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 3

id: 689
unit: def __init__()
file: optimum/exporters/neuron/model_wrappers.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 3

id: 690
unit: def forward()
file: optimum/exporters/neuron/model_wrappers.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 2

id: 691
unit: def outputs()
file: optimum/exporters/neuron/model_configs.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 2
number of parameters: 1

id: 692
unit: def outputs()
file: optimum/exporters/neuron/model_configs.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 2
number of parameters: 1

id: 693
unit: def outputs()
file: optimum/exporters/neuron/model_configs.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 2
number of parameters: 1

id: 694
unit: def outputs()
file: optimum/exporters/neuron/model_configs.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 2
number of parameters: 1

id: 695
unit: def is_lcm()
file: optimum/neuron/modeling_diffusion.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 2
number of parameters: 1

id: 696
unit: def _remove_submodel_if_non_exist()
file: optimum/neuron/modeling_diffusion.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 3
number of parameters: 1

id: 697
unit: def __init__()
file: optimum/neuron/modeling_diffusion.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 0

id: 698
unit: def __init__()
file: optimum/neuron/modeling_diffusion.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 1

id: 699
unit: def load_model()
file: optimum/neuron/modeling_seq2seq.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 0

id: 700
unit: def parameters_for_current_stage()
file: optimum/neuron/peft/peft_model.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 2
number of parameters: 1

id: 701
unit: def add_adapter()
file: optimum/neuron/peft/peft_model.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 2
number of parameters: 4

id: 702
unit: def __init__()
file: optimum/neuron/peft/tuners/lora/model.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 2
number of parameters: 5

id: 703
unit: def from_config()
file: optimum/neuron/generation/logits_process.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 4
number of parameters: 2

id: 704
unit: def _wrap_model()
file: optimum/neuron/trainers.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 4

id: 705
unit: def _prepare_input()
file: optimum/neuron/trainers.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 2
number of parameters: 3

id: 706
unit: def _load_from_checkpoint()
file: optimum/neuron/trainers.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 2
number of parameters: 3

id: 707
unit: def odds_ratio_loss()
file: optimum/neuron/trainers.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 0

id: 708
unit: def patch_generation_mixin_to_general_neuron_generation_mixin()
file: optimum/neuron/utils/training_utils.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 1

id: 709
unit: def requires_neuron_or_neuronx()
file: optimum/neuron/utils/testing_utils.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 2
number of parameters: 1

id: 710
unit: def create_custom_cache_repo()
file: optimum/neuron/utils/cache_utils.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 2

id: 711
unit: def convert_neuronx_compiler_args_to_neuron()
file: optimum/neuron/utils/argument_utils.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 0

id: 712
unit: def process_patching_specs()
file: optimum/neuron/utils/patching.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 0

id: 713
unit: def replace_weights()
file: optimum/neuron/utils/misc.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 0

id: 714
unit: def world_size()
file: optimum/neuron/training_args.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 2
number of parameters: 1

id: 715
unit: def replace_weights()
file: optimum/neuron/modeling_traced.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 2
number of parameters: 4

id: 716
unit: def _save_pretrained()
file: optimum/neuron/modeling_traced.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 3

id: 717
unit: def __init__()
file: optimum/neuron/models/training/qwen3/modeling_qwen3.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 4

id: 718
unit: def rotate_half()
file: optimum/neuron/models/training/llama/modeling_llama.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 1

id: 719
unit: def rotate_half()
file: optimum/neuron/models/training/granite/modeling_granite.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 1

id: 720
unit: def __init__()
file: optimum/neuron/models/training/granite/modeling_granite.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 4

id: 721
unit: def parameters_for_current_stage()
file: optimum/neuron/models/training/modeling_utils.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 2
number of parameters: 1

id: 722
unit: def _set_gradient_checkpointing()
file: optimum/neuron/models/training/modeling_utils.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 3

id: 723
unit: def is_leaf_module()
file: optimum/neuron/models/training/pipeline_utils.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 2
number of parameters: 3

id: 724
unit: def parallel_cross_entropy()
file: optimum/neuron/models/training/loss_utils.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 2
number of parameters: 5

id: 725
unit: def consolidate_model_parallel_checkpoints_to_unified_checkpoint()
file: optimum/neuron/models/training/checkpointing.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 0

id: 726
unit: def peft_type()
file: optimum/neuron/models/training/transformations_utils.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 2
number of parameters: 1

id: 727
unit: def add_spec()
file: optimum/neuron/models/training/transformations_utils.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 2
number of parameters: 2

id: 728
unit: def specs()
file: optimum/neuron/models/training/transformations_utils.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 2
number of parameters: 1

id: 729
unit: def specs()
file: optimum/neuron/models/training/transformations_utils.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 2
number of parameters: 2

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

id: 731
unit: def compile()
file: optimum/neuron/models/inference/backend/pretrained_model.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 4

id: 732
unit: def _save_pretrained()
file: optimum/neuron/models/inference/backend/modules/decoder/modeling_decoder.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 2
number of parameters: 3

id: 733
unit: def get_kv_by_layer_id()
file: optimum/neuron/models/inference/backend/modules/kvcache/kv_cache_manager.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 4

id: 734
unit: def __init__()
file: optimum/neuron/models/inference/backend/modules/custom_calls.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 3

id: 735
unit: def _rotate_half()
file: optimum/neuron/models/inference/backend/modules/attention/utils.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 1

id: 736
unit: def __init__()
file: optimum/neuron/models/inference/backend/model_wrapper.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 3

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

id: 738
unit: def prepare_optimizer()
file: optimum/neuron/accelerate/accelerator.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 2
number of parameters: 3

id: 739
unit: def forward()
file: optimum/neuron/modeling.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 0

id: 740
unit: def forward()
file: optimum/neuron/modeling.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 0

id: 741
unit: def forward()
file: optimum/neuron/modeling.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 0

id: 742
unit: def forward()
file: optimum/neuron/modeling.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 0

id: 743
unit: def forward()
file: optimum/neuron/modeling.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 0

id: 744
unit: def forward()
file: optimum/neuron/modeling.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 0

id: 745
unit: def forward()
file: optimum/neuron/modeling.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 0

id: 746
unit: def decode_hlo()
file: tools/decode_hlos.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 1

id: 747
unit: def _benchmark_rate_id()
file: benchmark/text-generation-inference/performance/generate_csv.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 2
number of parameters: 1

id: 748
unit: def main()
file: notebooks/text-classification/scripts/train.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 0

id: 749
unit: def run()
file: optimum/commands/export/neuron.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 1

id: 750
unit: def run()
file: optimum/commands/export/neuronx.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 1

id: 751
unit: def run()
file: optimum/commands/neuron/cache.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 1

id: 752
unit: def task()
file: optimum/exporters/neuron/base.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 2

id: 753
unit: def forward()
file: optimum/exporters/neuron/model_wrappers.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 3

id: 754
unit: def inputs()
file: optimum/exporters/neuron/model_configs.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 1

id: 755
unit: def generate_dummy_inputs()
file: optimum/exporters/neuron/model_configs.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 3

id: 756
unit: def _maybe_create_dummy_image_proj_layers()
file: optimum/neuron/modeling_diffusion.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 2
number of parameters: 1

id: 757
unit: def __init__()
file: optimum/neuron/modeling_diffusion.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 2

id: 758
unit: def __init__()
file: optimum/neuron/modeling_diffusion.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 2

id: 759
unit: def _save_pretrained()
file: optimum/neuron/modeling_seq2seq.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 3

id: 760
unit: def preprocess()
file: optimum/neuron/pipelines/transformers/sentence_transformers.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 3

id: 761
unit: def _forward()
file: optimum/neuron/pipelines/transformers/sentence_transformers.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 2

id: 762
unit: def __repr__()
file: optimum/neuron/peft/tuners/lora/layer.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 1

id: 763
unit: def __repr__()
file: optimum/neuron/peft/tuners/lora/layer.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 1

id: 764
unit: def __repr__()
file: optimum/neuron/peft/tuners/lora/layer.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 1

id: 765
unit: def _create_new_module()
file: optimum/neuron/peft/tuners/lora/model.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 4

id: 766
unit: def get_peft_model_state_dict()
file: optimum/neuron/peft/utils/save_and_load.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 2

id: 767
unit: def _prepare_inputs()
file: optimum/neuron/trainers.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 4

id: 768
unit: def set_verbosity()
file: optimum/neuron/utils/training_utils.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 1

id: 769
unit: def get_transformers_version()
file: optimum/neuron/utils/deprecate_utils.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 0

id: 770
unit: def split_args_and_value_in_command()
file: optimum/neuron/utils/runner.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 4
number of parameters: 1

id: 771
unit: def check_if_weights_replacable()
file: optimum/neuron/utils/misc.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 0

id: 772
unit: def to_dict()
file: optimum/neuron/utils/misc.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 1

id: 773
unit: def neuron_padding_manager()
file: optimum/neuron/modeling_traced.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 3

id: 774
unit: def __init__()
file: optimum/neuron/models/whisper/model.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 2
number of parameters: 2

id: 775
unit: def __init__()
file: optimum/neuron/models/whisper/model.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 3

id: 776
unit: def __init__()
file: optimum/neuron/models/training/qwen3/modeling_qwen3.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 4

id: 777
unit: def __init__()
file: optimum/neuron/models/training/qwen3/modeling_qwen3.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 3

id: 778
unit: def __init__()
file: optimum/neuron/models/training/granite/modeling_granite.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 4

id: 779
unit: def __enter__()
file: optimum/neuron/models/training/pipeline_utils.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 1

id: 780
unit: def __post_init__()
file: optimum/neuron/models/training/transformations_utils.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 2
number of parameters: 1

id: 781
unit: def __init__()
file: optimum/neuron/models/inference/qwen3/modeling_qwen3.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 3

id: 782
unit: def normalize_path()
file: optimum/neuron/models/inference/backend/pretrained_model.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 1

id: 783
unit: def _get_async_output()
file: optimum/neuron/models/inference/backend/modules/decoder/modeling_decoder.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 0

id: 784
unit: def load_state_dict()
file: optimum/neuron/models/inference/backend/modules/decoder/decoder_wrapper.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 4

id: 785
unit: def _get_async_output()
file: optimum/neuron/models/inference/backend/modules/decoder/decoder_wrapper.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 2
number of parameters: 2

id: 786
unit: def _is_using_pt2()
file: optimum/neuron/models/inference/backend/modules/checkpoint.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 0

id: 787
unit: def load_safetensors()
file: optimum/neuron/models/inference/backend/modules/checkpoint.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 1

id: 788
unit: def load_pytorch_model_bin()
file: optimum/neuron/models/inference/backend/modules/checkpoint.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 1

id: 789
unit: def _rand_selector()
file: optimum/neuron/models/inference/backend/modules/generation/sampling.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 2
number of parameters: 3

id: 790
unit: def _get_index_to_update_new_position()
file: optimum/neuron/models/inference/backend/modules/kvcache/kv_cache_manager.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 4

id: 791
unit: def forward()
file: optimum/neuron/models/inference/granite/modeling_granite.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 2

id: 792
unit: def forward()
file: optimum/neuron/models/inference/granite/modeling_granite.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 3

id: 793
unit: def has_neuron_model_class()
file: optimum/neuron/models/auto_model.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 4

id: 794
unit: def to_json_file()
file: optimum/neuron/configuration_utils.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 3

id: 795
unit: def patch_model_for_neuron()
file: optimum/neuron/accelerate/accelerator.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 0

id: 796
unit: def ls_hlos()
file: tools/decode_hlos.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 3
number of parameters: 1

id: 797
unit: def format_dict()
file: optimum/commands/env.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 2
number of parameters: 1

id: 798
unit: def parse_args()
file: optimum/commands/export/neuron.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 799
unit: def parse_args()
file: optimum/commands/export/neuronx.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 800
unit: def parse_args()
file: optimum/commands/neuron/cache.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 801
unit: def run()
file: optimum/commands/neuron/cache.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 802
unit: def run()
file: optimum/commands/neuron/cache.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 803
unit: def normalize_stable_diffusion_input_shapes()
file: optimum/exporters/neuron/__main__.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 804
unit: def get_diffusers_submodel_type()
file: optimum/exporters/neuron/utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 805
unit: def get_mandatory_axes_for_task()
file: optimum/exporters/neuron/base.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 806
unit: def task()
file: optimum/exporters/neuron/base.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 807
unit: def tensor_parallel_size()
file: optimum/exporters/neuron/base.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 808
unit: def tensor_parallel_size()
file: optimum/exporters/neuron/base.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 809
unit: def values_override()
file: optimum/exporters/neuron/base.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 810
unit: def inputs()
file: optimum/exporters/neuron/base.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 811
unit: def outputs()
file: optimum/exporters/neuron/base.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 812
unit: def generate_dummy_inputs()
file: optimum/exporters/neuron/base.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 813
unit: def __init__()
file: optimum/exporters/neuron/model_wrappers.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 814
unit: def forward()
file: optimum/exporters/neuron/model_wrappers.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 3

id: 815
unit: def inputs()
file: optimum/exporters/neuron/model_configs.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 816
unit: def inputs()
file: optimum/exporters/neuron/model_configs.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 817
unit: def inputs()
file: optimum/exporters/neuron/model_configs.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 818
unit: def inputs()
file: optimum/exporters/neuron/model_configs.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 819
unit: def inputs()
file: optimum/exporters/neuron/model_configs.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 820
unit: def inputs()
file: optimum/exporters/neuron/model_configs.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 821
unit: def inputs()
file: optimum/exporters/neuron/model_configs.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 822
unit: def outputs()
file: optimum/exporters/neuron/model_configs.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 823
unit: def inputs()
file: optimum/exporters/neuron/model_configs.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 824
unit: def inputs()
file: optimum/exporters/neuron/model_configs.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 825
unit: def outputs()
file: optimum/exporters/neuron/model_configs.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 826
unit: def inputs()
file: optimum/exporters/neuron/model_configs.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 827
unit: def inputs()
file: optimum/exporters/neuron/model_configs.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 828
unit: def outputs()
file: optimum/exporters/neuron/model_configs.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 829
unit: def is_sdxl()
file: optimum/exporters/neuron/model_configs.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 830
unit: def is_sdxl()
file: optimum/exporters/neuron/model_configs.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 831
unit: def with_controlnet()
file: optimum/exporters/neuron/model_configs.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 832
unit: def with_controlnet()
file: optimum/exporters/neuron/model_configs.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 833
unit: def with_ip_adapter()
file: optimum/exporters/neuron/model_configs.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 834
unit: def outputs()
file: optimum/exporters/neuron/model_configs.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 835
unit: def outputs()
file: optimum/exporters/neuron/model_configs.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 836
unit: def inputs()
file: optimum/exporters/neuron/model_configs.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 837
unit: def outputs()
file: optimum/exporters/neuron/model_configs.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 838
unit: def inputs()
file: optimum/exporters/neuron/model_configs.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 839
unit: def outputs()
file: optimum/exporters/neuron/model_configs.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 840
unit: def inputs()
file: optimum/exporters/neuron/model_configs.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 841
unit: def outputs()
file: optimum/exporters/neuron/model_configs.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 842
unit: def is_encoder_decoder()
file: optimum/exporters/neuron/model_configs.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 843
unit: def patch_model_for_export()
file: optimum/exporters/neuron/model_configs.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 3

id: 844
unit: def is_encoder_decoder()
file: optimum/exporters/neuron/model_configs.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 845
unit: def get_parallel_callable()
file: optimum/exporters/neuron/model_configs.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 846
unit: def is_encoder_decoder()
file: optimum/exporters/neuron/model_configs.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 847
unit: def inputs()
file: optimum/exporters/neuron/model_configs.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 848
unit: def outputs()
file: optimum/exporters/neuron/model_configs.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 849
unit: def is_encoder_decoder()
file: optimum/exporters/neuron/model_configs.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 850
unit: def patch_model_for_export()
file: optimum/exporters/neuron/model_configs.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 3

id: 851
unit: def inputs()
file: optimum/exporters/neuron/model_configs.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 852
unit: def outputs()
file: optimum/exporters/neuron/model_configs.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 853
unit: def is_encoder_decoder()
file: optimum/exporters/neuron/model_configs.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 854
unit: def patch_model_for_export()
file: optimum/exporters/neuron/model_configs.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 3

id: 855
unit: def _from_transformers()
file: optimum/neuron/modeling_diffusion.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 3

id: 856
unit: def _load_config()
file: optimum/neuron/modeling_diffusion.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 4

id: 857
unit: def _save_config()
file: optimum/neuron/modeling_diffusion.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 858
unit: def forward()
file: optimum/neuron/modeling_diffusion.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 3

id: 859
unit: def __call__()
file: optimum/neuron/modeling_diffusion.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 3

id: 860
unit: def dtype()
file: optimum/neuron/modeling_diffusion.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 861
unit: def to()
file: optimum/neuron/modeling_diffusion.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 3

id: 862
unit: def modules()
file: optimum/neuron/modeling_diffusion.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 2
number of parameters: 1

id: 863
unit: def config()
file: optimum/neuron/modeling_diffusion.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 864
unit: def encode()
file: optimum/neuron/modeling_diffusion.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 3

id: 865
unit: def decode()
file: optimum/neuron/modeling_diffusion.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 3

id: 866
unit: def __class__()
file: optimum/neuron/modeling_diffusion.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 867
unit: def __class__()
file: optimum/neuron/modeling_diffusion.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 868
unit: def __init__()
file: optimum/neuron/modeling_diffusion.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 869
unit: def forward()
file: optimum/neuron/modeling_seq2seq.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 3

id: 870
unit: def __call__()
file: optimum/neuron/modeling_seq2seq.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 3

id: 871
unit: def _from_transformers()
file: optimum/neuron/modeling_seq2seq.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 3

id: 872
unit: def _reorder_cache()
file: optimum/neuron/modeling_seq2seq.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 873
unit: def get_encoder()
file: optimum/neuron/modeling_seq2seq.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 874
unit: def _validate_static_shape()
file: optimum/neuron/modeling_seq2seq.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 3

id: 875
unit: def can_generate()
file: optimum/neuron/modeling_seq2seq.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 876
unit: def __call__()
file: optimum/neuron/pipelines/transformers/sentence_transformers.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 3

id: 877
unit: def check_model_type()
file: optimum/neuron/pipelines/transformers/base.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 3

id: 878
unit: def _get_add_time_ids_text_to_image()
file: optimum/neuron/pipelines/diffusers/pipeline_utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 879
unit: def get_peft_model_state_dict()
file: optimum/neuron/peft/peft_model.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 880
unit: def load_peft_weights()
file: optimum/neuron/peft/peft_model.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 881
unit: def set_peft_model_state_dict()
file: optimum/neuron/peft/peft_model.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 882
unit: def recompute_parameters_for_current_stage()
file: optimum/neuron/peft/peft_model.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 883
unit: def get_nb_trainable_parameters()
file: optimum/neuron/peft/peft_model.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 884
unit: def gather_params_ctx()
file: optimum/neuron/peft/tuners/lora/layer.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 885
unit: def __init__()
file: optimum/neuron/peft/tuners/lora/layer.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 3

id: 886
unit: def __init__()
file: optimum/neuron/peft/tuners/lora/layer.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 3

id: 887
unit: def __init__()
file: optimum/neuron/peft/tuners/lora/layer.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 3

id: 888
unit: def __init__()
file: optimum/neuron/peft/tuners/lora/layer.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 3

id: 889
unit: def __init__()
file: optimum/neuron/peft/tuners/lora/layer.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 3

id: 890
unit: def scatter_to_sequence_parallel_region()
file: optimum/neuron/peft/tuners/lora/layer.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 891
unit: def orig_get_peft_model_state_dict()
file: optimum/neuron/peft/utils/save_and_load.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 892
unit: def orig_get_peft_model()
file: optimum/neuron/peft/mapping.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 893
unit: def _pad_input_ids_for_general_sampling()
file: optimum/neuron/generation/utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 894
unit: def _prepare_encoder_decoder_kwargs_for_generation()
file: optimum/neuron/generation/utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 895
unit: def mp_enabled()
file: optimum/neuron/trainers.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 896
unit: def _get_eval_sampler()
file: optimum/neuron/trainers.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 897
unit: def get_num_trainable_parameters()
file: optimum/neuron/trainers.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 898
unit: def training_step()
file: optimum/neuron/trainers.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 899
unit: def _inner_training_loop()
file: optimum/neuron/trainers.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 900
unit: def predict()
file: optimum/neuron/trainers.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 901
unit: def save_metrics()
file: optimum/neuron/trainers.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 4

id: 902
unit: def save_state()
file: optimum/neuron/trainers.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 903
unit: def __init__()
file: optimum/neuron/trainers.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 3

id: 904
unit: def make_inputs_require_grad()
file: optimum/neuron/trainers.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 3

id: 905
unit: def __init__()
file: optimum/neuron/trainers.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 3

id: 906
unit: def make_inputs_require_grad()
file: optimum/neuron/trainers.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 3

id: 907
unit: def make_inputs_require_grad()
file: optimum/neuron/trainers.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 3

id: 908
unit: def concatenated_forward()
file: optimum/neuron/trainers.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 909
unit: def log1p()
file: optimum/neuron/trainers.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 910
unit: def is_neuron_available()
file: optimum/neuron/utils/import_utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 911
unit: def is_neuronx_available()
file: optimum/neuron/utils/import_utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 912
unit: def is_neuronx_distributed_available()
file: optimum/neuron/utils/import_utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 913
unit: def is_torch_neuronx_available()
file: optimum/neuron/utils/import_utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 914
unit: def patch_generation_mixin_to_neuron_generation_mixin()
file: optimum/neuron/utils/training_utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 915
unit: def patch_transformers_for_neuron_sdk()
file: optimum/neuron/utils/training_utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 916
unit: def is_main_worker_for_metrics_method()
file: optimum/neuron/utils/training_utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 917
unit: def __init__()
file: optimum/neuron/utils/trl_utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 3

id: 918
unit: def __init__()
file: optimum/neuron/utils/trl_utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 3

id: 919
unit: def neuron_cc_flags_model_type()
file: optimum/neuron/utils/trl_utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 920
unit: def requires_neuron()
file: optimum/neuron/utils/testing_utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 921
unit: def requires_neuronx()
file: optimum/neuron/utils/testing_utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 922
unit: def load_custom_cache_repo_name_from_hf_home()
file: optimum/neuron/utils/cache_utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 923
unit: def set_custom_cache_repo_name_in_hf_home()
file: optimum/neuron/utils/cache_utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 924
unit: def delete_custom_cache_repo_name_from_hf_home()
file: optimum/neuron/utils/cache_utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 925
unit: def get_hf_hub_cache_repo()
file: optimum/neuron/utils/cache_utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 926
unit: def get_num_neuron_cores_used()
file: optimum/neuron/utils/cache_utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 927
unit: def list_filenames_in_github_repo_directory()
file: optimum/neuron/utils/runner.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 928
unit: def maybe_remove_venv()
file: optimum/neuron/utils/runner.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 929
unit: def download_model_repo_and_override_config()
file: optimum/neuron/utils/runner.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 930
unit: def expected_validation_function()
file: optimum/neuron/utils/argument_utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 931
unit: def __init__()
file: optimum/neuron/utils/patching.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 932
unit: def process_patching_specs()
file: optimum/neuron/utils/patching.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 933
unit: def __enter__()
file: optimum/neuron/utils/patching.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 934
unit: def __exit__()
file: optimum/neuron/utils/patching.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 4

id: 935
unit: def __init__()
file: optimum/neuron/utils/patching.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 3

id: 936
unit: def __call__()
file: optimum/neuron/utils/patching.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 937
unit: def process_patching_specs()
file: optimum/neuron/utils/patching.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 938
unit: def patch_within_function()
file: optimum/neuron/utils/patching.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 939
unit: def is_precompilation()
file: optimum/neuron/utils/misc.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 940
unit: def _is_weights_neff_separated()
file: optimum/neuron/utils/misc.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 2
number of parameters: 1

id: 941
unit: def get_stable_diffusion_configs()
file: optimum/neuron/utils/misc.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 942
unit: def get_parent_module_and_param_name_from_fully_qualified_name()
file: optimum/neuron/utils/model_utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 943
unit: def _setup_devices()
file: optimum/neuron/training_args.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 944
unit: def neuron_cc_flags_model_type()
file: optimum/neuron/training_args.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 945
unit: def place_model_on_device()
file: optimum/neuron/training_args.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 2
number of parameters: 1

id: 946
unit: def world_size_should_behave_as_dp_size()
file: optimum/neuron/training_args.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 947
unit: def load_model()
file: optimum/neuron/modeling_traced.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 948
unit: def _from_transformers()
file: optimum/neuron/modeling_traced.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 3

id: 949
unit: def forward()
file: optimum/neuron/modeling_traced.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 3

id: 950
unit: def _pad_to_compiled_shape()
file: optimum/neuron/modeling_traced.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 951
unit: def is_weights_neff_separated()
file: optimum/neuron/modeling_traced.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 952
unit: def can_generate()
file: optimum/neuron/modeling_traced.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 953
unit: def __call__()
file: optimum/neuron/models/whisper/model.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 954
unit: def device()
file: optimum/neuron/models/whisper/model.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 955
unit: def get_encoder()
file: optimum/neuron/models/whisper/model.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 956
unit: def _init_normal()
file: optimum/neuron/models/training/qwen3/modeling_qwen3.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 957
unit: def should_parallelize()
file: optimum/neuron/models/training/config.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 2
number of parameters: 1

id: 958
unit: def _init_normal()
file: optimum/neuron/models/training/llama/modeling_llama.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 959
unit: def extra_repr()
file: optimum/neuron/models/training/llama/modeling_llama.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 960
unit: def get_input_embeddings()
file: optimum/neuron/models/training/llama/modeling_llama.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 961
unit: def set_input_embeddings()
file: optimum/neuron/models/training/llama/modeling_llama.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 962
unit: def get_input_embeddings()
file: optimum/neuron/models/training/llama/modeling_llama.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 963
unit: def set_input_embeddings()
file: optimum/neuron/models/training/llama/modeling_llama.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 964
unit: def get_output_embeddings()
file: optimum/neuron/models/training/llama/modeling_llama.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 965
unit: def set_output_embeddings()
file: optimum/neuron/models/training/llama/modeling_llama.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 966
unit: def set_decoder()
file: optimum/neuron/models/training/llama/modeling_llama.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 967
unit: def get_decoder()
file: optimum/neuron/models/training/llama/modeling_llama.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 968
unit: def checkpoint()
file: optimum/neuron/models/training/modeling_utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 969
unit: def nki_flash_attn_func()
file: optimum/neuron/models/training/modeling_utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 970
unit: def __init__()
file: optimum/neuron/models/training/modeling_utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 3

id: 971
unit: def __init__()
file: optimum/neuron/models/training/modeling_utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 3

id: 972
unit: def __init__()
file: optimum/neuron/models/training/modeling_utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 3

id: 973
unit: def __init__()
file: optimum/neuron/models/training/modeling_utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 3

id: 974
unit: def get_data_parallel_rank()
file: optimum/neuron/models/training/modeling_utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 975
unit: def get_pipeline_model_parallel_rank()
file: optimum/neuron/models/training/modeling_utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 976
unit: def get_pipeline_model_parallel_size()
file: optimum/neuron/models/training/modeling_utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 977
unit: def get_tensor_model_parallel_rank()
file: optimum/neuron/models/training/modeling_utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 978
unit: def get_tensor_model_parallel_size()
file: optimum/neuron/models/training/modeling_utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 979
unit: def get_local_world_size()
file: optimum/neuron/models/training/modeling_utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 980
unit: def move_model_to_device()
file: optimum/neuron/models/training/modeling_utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 981
unit: def get_pipeline_model_parallel_size()
file: optimum/neuron/models/training/pipeline_utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 982
unit: def __init__()
file: optimum/neuron/models/training/pipeline_utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 3

id: 983
unit: def __init__()
file: optimum/neuron/models/training/pipeline_utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 3

id: 984
unit: def __init__()
file: optimum/neuron/models/training/pipeline_utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 3

id: 985
unit: def __exit__()
file: optimum/neuron/models/training/pipeline_utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 4

id: 986
unit: def dynamic_torch_fx_wrap()
file: optimum/neuron/models/training/pipeline_utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 987
unit: def ForCausalLMLoss()
file: optimum/neuron/models/training/loss_utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 988
unit: def consolidate_model_parallel_checkpoints()
file: optimum/neuron/models/training/checkpointing.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 989
unit: def get_tensor_model_parallel_size()
file: optimum/neuron/models/training/transformations_utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 990
unit: def peft_type()
file: optimum/neuron/models/training/transformations_utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 991
unit: def get_relevant_parameter_names()
file: optimum/neuron/models/training/transformations_utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 992
unit: def guess_peft_type()
file: optimum/neuron/models/training/transformations_utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 3

id: 993
unit: def adapt_peft_config()
file: optimum/neuron/models/training/transformations_utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 3

id: 994
unit: def to_original_peft_config()
file: optimum/neuron/models/training/transformations_utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 3

id: 995
unit: def __iter__()
file: optimum/neuron/models/training/transformations_utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 996
unit: def is_transformation_spec_relevant()
file: optimum/neuron/models/training/transformations_utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 997
unit: def to_original_weights()
file: optimum/neuron/models/training/transformations_utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 998
unit: def __repr__()
file: optimum/neuron/models/training/transformations_utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 999
unit: def compute_query_indices_for_rank()
file: optimum/neuron/models/training/transformations_utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 1000
unit: def create_kv_proj_local_weight_from_regular_weight()
file: optimum/neuron/models/training/transformations_utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 1001
unit: def adapt_peft_config_for_model()
file: optimum/neuron/models/training/transformations_utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 1002
unit: def to_original_peft_config_for_model()
file: optimum/neuron/models/training/transformations_utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 1003
unit: def remove_adapter_name()
file: optimum/neuron/models/training/transformations_utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 1004
unit: def is_base_layer()
file: optimum/neuron/models/training/transformations_utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 1005
unit: def should_keep_parameter()
file: optimum/neuron/models/training/transformations_utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 2
number of parameters: 1

id: 1006
unit: def dtype()
file: optimum/neuron/models/yolos/model.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 1007
unit: def get_shards_path()
file: optimum/neuron/models/inference/backend/pretrained_model.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 1008
unit: def forward()
file: optimum/neuron/models/inference/backend/pretrained_model.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 1009
unit: def get_config_cls()
file: optimum/neuron/models/inference/backend/pretrained_model.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 1010
unit: def get_neuron_config_cls()
file: optimum/neuron/models/inference/backend/pretrained_model.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 1011
unit: def get_compiler_args()
file: optimum/neuron/models/inference/backend/pretrained_model.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 1012
unit: def convert_hf_to_neuron_state_dict()
file: optimum/neuron/models/inference/backend/pretrained_model.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 1013
unit: def load_hf_model()
file: optimum/neuron/models/inference/backend/pretrained_model.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 1014
unit: def update_state_dict_for_tied_weights()
file: optimum/neuron/models/inference/backend/pretrained_model.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 1015
unit: def device()
file: optimum/neuron/models/inference/backend/pretrained_model.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 1016
unit: def reset()
file: optimum/neuron/models/inference/backend/pretrained_model.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 1017
unit: def ctx_batch_size()
file: optimum/neuron/models/inference/backend/config.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 2
number of parameters: 1

id: 1018
unit: def tkg_batch_size()
file: optimum/neuron/models/inference/backend/config.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 1019
unit: def world_size()
file: optimum/neuron/models/inference/backend/config.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 1020
unit: def weights_to_skip_layout_optimization()
file: optimum/neuron/models/inference/backend/config.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 1021
unit: def _is_context_encoding()
file: optimum/neuron/models/inference/backend/modules/decoder/modeling_decoder.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 2
number of parameters: 2

id: 1022
unit: def _is_for_speculation()
file: optimum/neuron/models/inference/backend/modules/decoder/modeling_decoder.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 1023
unit: def _create_simple_attn_mask()
file: optimum/neuron/models/inference/backend/modules/decoder/modeling_decoder.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 1024
unit: def _is_reorder_needed()
file: optimum/neuron/models/inference/backend/modules/decoder/modeling_decoder.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 3
number of parameters: 3

id: 1025
unit: def get_input_embeddings()
file: optimum/neuron/models/inference/backend/modules/decoder/modeling_decoder.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 1026
unit: def set_input_embeddings()
file: optimum/neuron/models/inference/backend/modules/decoder/modeling_decoder.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 1027
unit: def create_token_generation_wrapper()
file: optimum/neuron/models/inference/backend/modules/decoder/modeling_decoder.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 1028
unit: def reset()
file: optimum/neuron/models/inference/backend/modules/decoder/modeling_decoder.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 2
number of parameters: 1

id: 1029
unit: def get_required_kwargs()
file: optimum/neuron/models/inference/backend/modules/decoder/modeling_decoder.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 1030
unit: def get_bucket_model_config_from_tag()
file: optimum/neuron/models/inference/backend/modules/decoder/decoder_wrapper.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 1031
unit: def input_generator()
file: optimum/neuron/models/inference/backend/modules/decoder/decoder_wrapper.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 1032
unit: def get_bucket_config()
file: optimum/neuron/models/inference/backend/modules/decoder/decoder_wrapper.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 1033
unit: def convert_int64_to_int32()
file: optimum/neuron/models/inference/backend/modules/decoder/decoder_wrapper.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 3
number of parameters: 2

id: 1034
unit: def __init__()
file: optimum/neuron/models/inference/backend/modules/decoder/decoder_wrapper.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 1035
unit: def initialize_process_group()
file: optimum/neuron/models/inference/backend/modules/decoder/decoder_wrapper.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 1036
unit: def generation_model_bk()
file: optimum/neuron/models/inference/backend/modules/autobucketing.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 1037
unit: def get_generation_model_bk()
file: optimum/neuron/models/inference/backend/modules/autobucketing.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 1038
unit: def get_context_encoder_bk()
file: optimum/neuron/models/inference/backend/modules/autobucketing.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 1039
unit: def get_cache_size()
file: optimum/neuron/models/inference/backend/modules/flashdecode/utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 1040
unit: def turn_2d_mask_to_4d()
file: optimum/neuron/models/inference/backend/modules/flashdecode/utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 3

id: 1041
unit: def mask_util()
file: optimum/neuron/models/inference/backend/modules/flashdecode/utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 1042
unit: def __init__()
file: optimum/neuron/models/inference/backend/modules/generation/sampling.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 1043
unit: def _soft_max()
file: optimum/neuron/models/inference/backend/modules/generation/sampling.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 3

id: 1044
unit: def can_generate()
file: optimum/neuron/models/inference/backend/modules/generation/generation_utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 1045
unit: def device()
file: optimum/neuron/models/inference/backend/modules/generation/generation_utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 1046
unit: def determine_sharding_strategy()
file: optimum/neuron/models/inference/backend/modules/attention/gqa.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 1047
unit: def get_shardable_head_counts()
file: optimum/neuron/models/inference/backend/modules/attention/gqa.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 1048
unit: def get_sharding_strategy()
file: optimum/neuron/models/inference/backend/modules/attention/gqa.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 1049
unit: def get_num_attention_heads()
file: optimum/neuron/models/inference/backend/modules/attention/gqa.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 1050
unit: def get_num_key_value_heads()
file: optimum/neuron/models/inference/backend/modules/attention/gqa.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 1051
unit: def preshard_hook()
file: optimum/neuron/models/inference/backend/modules/attention/gqa.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 3

id: 1052
unit: def get_weight()
file: optimum/neuron/models/inference/backend/modules/attention/gqa.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 1053
unit: def get_bias()
file: optimum/neuron/models/inference/backend/modules/attention/gqa.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 1054
unit: def forward()
file: optimum/neuron/models/inference/backend/modules/attention/gqa.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 1055
unit: def input_generator()
file: optimum/neuron/models/inference/backend/model_wrapper.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 1056
unit: def get_model_instance()
file: optimum/neuron/models/inference/backend/model_wrapper.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 1057
unit: def get_bucket_config()
file: optimum/neuron/models/inference/backend/model_wrapper.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 1058
unit: def forward()
file: optimum/neuron/models/inference/backend/model_wrapper.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 1059
unit: def register_neuron_model_for_inference()
file: optimum/neuron/models/inference/auto_models.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 1060
unit: def update_state_dict_for_tied_weights()
file: optimum/neuron/models/inference/llama/modeling_llama.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 1061
unit: def get_neuron_config_cls()
file: optimum/neuron/models/inference/llama/modeling_llama.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 1062
unit: def get_rmsnorm_cls()
file: optimum/neuron/models/inference/mixtral/modeling_mixtral.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 1063
unit: def get_neuron_config_cls()
file: optimum/neuron/models/inference/mixtral/modeling_mixtral.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 1064
unit: def convert_hf_to_neuron_state_dict()
file: optimum/neuron/models/inference/mixtral/modeling_mixtral.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 1065
unit: def dtype()
file: optimum/neuron/models/clip/model.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 1066
unit: def _from_transformers()
file: optimum/neuron/modeling_decoder.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 3

id: 1067
unit: def __hash__()
file: optimum/neuron/configuration_utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 1068
unit: def __repr__()
file: optimum/neuron/configuration_utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 1069
unit: def wait_for_everyone()
file: optimum/neuron/accelerate/state.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 1070
unit: def autocast_backend()
file: optimum/neuron/accelerate/state.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 1071
unit: def load_state_dict()
file: optimum/neuron/accelerate/optimizer.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 1072
unit: def prepare_clip_grad_norm()
file: optimum/neuron/accelerate/optimizer.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 4

id: 1073
unit: def __init__()
file: optimum/neuron/accelerate/utils/misc.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 3

id: 1074
unit: def patched_accelerate_is_torch_xla_available()
file: optimum/neuron/accelerate/utils/misc.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 1075
unit: def patched_is_torch_xla_available()
file: optimum/neuron/accelerate/accelerator.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 1076
unit: def prepare_data_loader()
file: optimum/neuron/accelerate/accelerator.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 1077
unit: def prepare_scheduler()
file: optimum/neuron/accelerate/accelerator.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 1078
unit: def prepare_model()
file: optimum/neuron/accelerate/accelerator.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 1079
unit: def _inner()
file: optimum/neuron/accelerate/accelerator.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 1080
unit: def save_state()
file: optimum/neuron/accelerate/accelerator.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 1081
unit: def gather()
file: optimum/neuron/accelerate/accelerator.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 3

id: 1082
unit: def dtype()
file: optimum/neuron/modeling.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 1083
unit: def dtype()
file: optimum/neuron/modeling.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 1084
unit: def dtype()
file: optimum/neuron/modeling.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 1085
unit: def build_stable_diffusion_command()
file: tools/auto_fill_inference_cache.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 1086
unit: def build_pixart_command()
file: tools/auto_fill_inference_cache.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 1087
unit: def predicate()
file: tools/create_examples_from_transformers.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 1088
unit: def insert_code_at_position()
file: tools/create_examples_from_transformers.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 3

id: 1089
unit: def remove_extra_command_line_argument()
file: tools/auto_fill_neuron_cache.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0