Searched defs:prev_flat_grad (Results 1 – 2 of 2) sorted by relevance
468 auto& prev_flat_grad = state.prev_flat_grad(); in step() local632 Tensor d, t, H_diag, prev_flat_grad, prev_loss; in load() local
842 at::Tensor d, t, H_diag, prev_flat_grad, prev_loss; in TEST() local