Searched refs:_fused_adam_kernel_cpu_ (Results 1 – 1 of 1) sorted by relevance
19 void _fused_adam_kernel_cpu_( in _fused_adam_kernel_cpu_() function76 void _fused_adam_kernel_cpu_( in _fused_adam_kernel_cpu_() function92 …_fused_adam_kernel_cpu_(params, grads, exp_avgs, exp_avg_sqs, max_exp_avg_sqs, state_steps, lr.ite… in _fused_adam_kernel_cpu_()