Searched defs:state_sum (Results 1 – 1 of 1) sorted by relevance
143 const at::Tensor& state_sum, in adagrad_fused_step_impl()190 const at::Tensor& state_sum, in fused_adagrad_kernel()