Searched defs:oStrideH (Results 1 – 2 of 2) sorted by relevance
226 int64_t oStrideH = output.stride(2); in cpu_flash_attention() local469 int64_t oStrideH = out.stride(2); in cpu_flash_attention_backward() local
340 int64_t oStrideH = strides[1]; in cpu_flash_attention() local