Lines Matching defs:Tensor

36         lr: Union[float, Tensor] = 1e-3,  argument
318     params: List[Tensor],
319     grads: List[Tensor],
320     exp_avgs: List[Tensor],
321     exp_avg_sqs: List[Tensor],
322     max_exp_avg_sqs: List[Tensor],
323     state_steps: List[Tensor],
324     grad_scale: Optional[Tensor],
325     found_inf: Optional[Tensor],
330     lr: Union[Tensor, float],
437     params: List[Tensor],
438     grads: List[Tensor],
439     exp_avgs: List[Tensor],
440     exp_avg_sqs: List[Tensor],
441     max_exp_avg_sqs: List[Tensor],
442     state_steps: List[Tensor],
443     grad_scale: Optional[Tensor],
444     found_inf: Optional[Tensor],
449     lr: Union[Tensor, float],
619     params: List[Tensor],
620     grads: List[Tensor],
621     exp_avgs: List[Tensor],
622     exp_avg_sqs: List[Tensor],
623     max_exp_avg_sqs: List[Tensor],
624     state_steps: List[Tensor],
625     grad_scale: Optional[Tensor],
626     found_inf: Optional[Tensor],
631     lr: Union[Tensor, float],
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,