Lines Matching defs:Tensor

31         lr: Union[float, Tensor] = 1.0,  argument
86         params_with_grad: List[Tensor],
87         grads: List[Tensor],
88         square_avgs: List[Tensor],
89         acc_deltas: List[Tensor],
90         state_steps: List[Tensor],
246     params: List[Tensor],
247     grads: List[Tensor],
248     square_avgs: List[Tensor],
249     acc_deltas: List[Tensor],
250     state_steps: List[Tensor],
300     params: List[Tensor],
301     grads: List[Tensor],
302     square_avgs: List[Tensor],
303     acc_deltas: List[Tensor],
304     state_steps: List[Tensor],
401     params: List[Tensor],
402     grads: List[Tensor],
403     square_avgs: List[Tensor],
404     acc_deltas: List[Tensor],
405     state_steps: List[Tensor],