Lines Matching refs:input_kwargs

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))
410             inputs = [(input_args, input_kwargs), (nc_input_args, nc_input_kwargs)]
460 …      input_args, input_kwargs = module_input.forward_input.args, module_input.forward_input.kwargs
463                     m(*input_args, **input_kwargs)
468             for name, obj in input_kwargs.items():