Searched defs:ModelArgs (Results 1 – 6 of 6) sorted by relevance
| /aosp_15_r20/external/executorch/examples/models/llama/ |
| H A D | llama_transformer.py | 81 class ModelArgs: class
265 def __init__(self, args: ModelArgs, layer_id: int):
364 def __init__(self, args: ModelArgs):
377 def __init__(self, args: ModelArgs):
423 def __init__(self, layer_id: int, args: ModelArgs):
451 def __init__(self, params: ModelArgs):
|
| /aosp_15_r20/external/pytorch/benchmarks/gpt_fast/ |
| H A D | model.py | 18 class ModelArgs: class
113 def __init__(self, config: ModelArgs) -> None:
171 def __init__(self, config: ModelArgs) -> None:
187 def __init__(self, config: ModelArgs):
245 def __init__(self, config: ModelArgs) -> None:
|
| H A D | mixtral_moe_model.py | 18 class ModelArgs: class
92 def __init__(self, config: ModelArgs) -> None:
150 def __init__(self, config: ModelArgs) -> None:
166 def __init__(self, config: ModelArgs):
|
| /aosp_15_r20/external/executorch/examples/qualcomm/oss_scripts/llama2/model/ |
| H A D | static_llama.py | 46 def __init__(self, config: ModelArgs, output_new_cache_only=False):
186 def __init__(self, config: ModelArgs, output_new_cache_only=False):
219 def __init__(self, config: ModelArgs, output_new_cache_only=True):
|
| /aosp_15_r20/external/pytorch/torch/testing/_internal/distributed/_tensor/ |
| H A D | common_dtensor.py | 92 class ModelArgs: class
105 def __init__(self, args: ModelArgs):
155 def __init__(self, args: ModelArgs):
173 def __init__(self, args: ModelArgs):
|
| /aosp_15_r20/external/pytorch/test/distributed/_tools/ |
| H A D | test_runtime_estimator.py | 107 model_args: Union[ConvArgs, ModelArgs], argument
|