Searched defs:kStrideB (Results 1 – 2 of 2) sorted by relevance
218 int64_t kStrideB = key.stride(0); in cpu_flash_attention() local461 int64_t kStrideB = key.stride(0); in cpu_flash_attention_backward() local
319 int64_t kStrideB = strides[0]; in cpu_flash_attention() local