Searched defs:s_r (Results 1 – 4 of 4) sorted by relevance
147 s_r = 1 variable207 s_r = 0.9 * s_r + 0.1 * scale[-1].detach().std().cpu().item() variable
311 s_r = 1 variable373 s_r = 0.9 * s_r + 0.1 * scale[-1].detach().std().cpu().item() variable
309 s_r = 1 variable375 s_r = 0.9 * s_r + 0.1 * score[-1].detach().std().cpu().item() variable
236 for (int64_t s_r = 0; s_r < shard_rows; ++s_r) { in operator ()() local327 for (int64_t s_r = 0; s_r < shard_rows; ++s_r) { in operator ()() local