Searched defs:state_sums (Results 1 – 2 of 2) sorted by relevance
20 at::TensorList state_sums, in _fused_adagrad_kernel_cpu_()
123 def _init_group(self, group, params_with_grad, grads, state_sums, state_steps): argument