Searched defs:state_step (Results 1 – 2 of 2) sorted by relevance
144 const at::Tensor& state_step, in adagrad_fused_step_impl()191 const at::Tensor& state_step, in fused_adagrad_kernel()
272 const at::Tensor& state_step, in adam_fused_step_impl()343 const at::Tensor& state_step, in fused_adam_kernel()