Home
last modified time | relevance | path

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

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