Home
last modified time | relevance | path

Searched defs:total_row (Results 1 – 1 of 1) sorted by relevance

/external/pytorch/aten/src/ATen/native/transformers/cuda/mem_eff_attention/
Dkernel_forward.h1286 accum_t mi_row, total_row; in iterative_softmax() local
1315 accum_t total_row = s_prime[id]; in iterative_softmax() local