Searched defs:model_args (Results 1 – 7 of 7) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/onnx/_internal/exporter/ |
| H A D | _onnx_program.py | 255 def _flatten_inputs(model_args, model_kwargs): argument
260 def _remove_none_from_inputs(model_args): argument
264 def _remove_non_tensor(model_args): argument
305 def _convert_complex_to_real_representation(model_args): argument
|
| /aosp_15_r20/external/pytorch/torch/onnx/_internal/ |
| H A D | _exporter_legacy.py | 614 *model_args,
1132 *model_args,
|
| H A D | io_adapter.py | 62 *model_args,
|
| /aosp_15_r20/external/executorch/examples/models/llama/ |
| H A D | model.py | 269 def _transform_for_pre_quantization(self, checkpoint, model_args): argument
|
| /aosp_15_r20/external/pytorch/torch/onnx/_internal/fx/ |
| H A D | fx_symbolic_graph_extractor.py | 162 self, model, model_args, model_kwargs argument
|
| /aosp_15_r20/external/pytorch/torch/onnx/ |
| H A D | __init__.py | 400 *model_args,
|
| /aosp_15_r20/external/pytorch/test/distributed/tensor/parallel/ |
| H A D | test_tp_examples.py | 201 def _setup_single_gpu_model(self, model_args, dtype): argument
|