Searched refs:input_kwargs (Results 1 – 9 of 9) sorted by relevance
| /external/pytorch/test/onnx/ |
| D | onnx_test_common.py | 150 input_kwargs=None, argument
168 input_kwargs=input_kwargs,
211 input_kwargs: Optional[Mapping[str, _InputArgsType]] = None,
251 if input_kwargs is None:
252 input_kwargs = {}
261 input_args, input_kwargs
266 ref_input_kwargs = input_kwargs
311 input_kwargs,
397 def _try_clone_inputs(input_args, input_kwargs): argument
399 ref_input_kwargs = copy.deepcopy(input_kwargs)
[all …]
|
| D | test_fx_to_onnx_with_onnxruntime.py | 138 func, (tensor_x,), input_kwargs={"b": torch.tensor(5.0)}
693 input_kwargs={
|
| /external/pytorch/test/onnx/torch_export/ |
| D | test_torch_export_with_onnxruntime.py | 23 input_kwargs=None, argument
28 if input_kwargs is None:
29 input_kwargs = {}
34 onnx_outputs = onnx_exported_program(*input_args, **input_kwargs)
36 torch_outputs = torch_exported_program.module()(*input_args, **input_kwargs)
38 torch_outputs = torch_exported_program(*input_args, **input_kwargs)
314 exported_program, onnx_program, input_args=(), input_kwargs=inputs
|
| /external/pytorch/test/ |
| D | test_modules.py | 281 … input_args, input_kwargs = module_input.forward_input.args, module_input.forward_input.kwargs
287 self.assertNotIn(first_param_name, input_kwargs)
293 output_op = m_op(*input_args, **input_kwargs)
301 output_ip = m_inplace(*input_arg_clone, **input_kwargs)
375 … input_args, input_kwargs = module_input.forward_input.args, module_input.forward_input.kwargs
376 if not (_can_be_noncontiguous(input_args) or _can_be_noncontiguous(input_kwargs)):
385 self._retain_grad((input_args, input_kwargs))
389 default_output = m(*input_args, **input_kwargs)
402 …_input_args_grad, default_input_kwargs_grad = deepcopy(self._get_grads((input_args, input_kwargs)))
406 nc_input_args, nc_input_kwargs = _make_non_contiguous((input_args, input_kwargs))
[all …]
|
| /external/pytorch/torch/onnx/ |
| D | verification.py | 386 input_kwargs: _InputKwargsType | None,
407 def compare_onnx_pytorch_model_with_input(input_args, input_kwargs): argument
408 pt_args, pt_kwargs = _prepare_input_for_pytorch(input_args, input_kwargs)
414 input_args, input_kwargs, options.remained_onnx_input_idx, options.flatten
425 compare_onnx_pytorch_model_with_input(input_args, input_kwargs)
775 input_kwargs: _InputKwargsType | None = None,
830 inputs_for_export = _prepare_input_for_export(input_args, input_kwargs)
853 input_kwargs=input_kwargs,
|
| D | utils.py | 1020 input_kwargs = {}
1022 input_kwargs = input_args[-1]
1025 example_outputs = model(*input_args, **input_kwargs)
|
| /external/pytorch/torchgen/api/ |
| D | python.py | 346 input_kwargs: tuple[PythonArgument, ...]
373 result.extend(self.input_kwargs)
384 return len(self.input_args) + len(self.input_kwargs)
774 input_kwargs = tuple(
879 input_kwargs=input_kwargs,
|
| /external/pytorch/tools/autograd/ |
| D | gen_python_functions.py | 671 input_kwargs=python_sig.input_kwargs,
|
| /external/pytorch/test/jit/ |
| D | test_tracer.py | 621 input_kwargs = kwargs.copy()
622 if "out" in input_kwargs:
623 del input_kwargs["out"]
624 input = torch.ones(2, 3, **input_kwargs)
|