Searched defs:H_diag (Results 1 – 2 of 2) sorted by relevance
467 auto& H_diag = state.H_diag(); 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