Home
last modified time | relevance | path

Searched defs:model_kwargs (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/pytorch/torch/onnx/_internal/
H A D_exporter_legacy.py616 **model_kwargs,
1134 **model_kwargs,
H A Dio_adapter.py64 **model_kwargs,
/aosp_15_r20/external/pytorch/torch/onnx/_internal/fx/
H A Dfx_symbolic_graph_extractor.py162 self, model, model_args, model_kwargs argument
/aosp_15_r20/external/pytorch/torch/onnx/_internal/exporter/
H A D_onnx_program.py255 def _flatten_inputs(model_args, model_kwargs): argument
/aosp_15_r20/external/pytorch/torch/onnx/
H A D__init__.py402 **model_kwargs,
/aosp_15_r20/external/pytorch/test/dynamo/
H A Dtest_repros.py548 def _validate_model_kwargs(fn, model_kwargs): argument