Lines Matching defs:Tensor

36         lr: Union[float, Tensor] = 1e-3,  argument
321     params: List[Tensor],
322     grads: List[Tensor],
323     exp_avgs: List[Tensor],
324     exp_avg_sqs: List[Tensor],
325     max_exp_avg_sqs: List[Tensor],
326     state_steps: List[Tensor],
327     grad_scale: Optional[Tensor],
328     found_inf: Optional[Tensor],
440     params: List[Tensor],
441     grads: List[Tensor],
442     exp_avgs: List[Tensor],
443     exp_avg_sqs: List[Tensor],
444     max_exp_avg_sqs: List[Tensor],
445     state_steps: List[Tensor],
446     grad_scale: Optional[Tensor],
447     found_inf: Optional[Tensor],
621     params: List[Tensor],
622     grads: List[Tensor],
623     exp_avgs: List[Tensor],
624     exp_avg_sqs: List[Tensor],
625     max_exp_avg_sqs: List[Tensor],
626     state_steps: List[Tensor],
627     grad_scale: Optional[Tensor],
628     found_inf: Optional[Tensor],
719     params: List[Tensor],
720     grads: List[Tensor],
721     exp_avgs: List[Tensor],
722     exp_avg_sqs: List[Tensor],
723     max_exp_avg_sqs: List[Tensor],
724     state_steps: List[Tensor],
731     grad_scale: Optional[Tensor] = None,
732     found_inf: Optional[Tensor] = None,