Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/native/nested/cuda/
H A DNestedTensorTransformerUtils.cpp268 int64_t max_seqlen_batch_kv = 0, Nnz_kv = 0; in sdpa_nested_preprocessing_with_broadcast() local
467 const int64_t max_seqlen_batch_kv) { in sdpa_nested_preprocessing_backward()