Home
last modified time | relevance | path

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

/external/pytorch/aten/src/ATen/native/nested/cuda/
DNestedTensorTransformerUtils.cpp250 int64_t max_seqlen_batch_q = 0, Nnz_q = 0; in sdpa_nested_preprocessing_with_broadcast() local
492 const int64_t Nnz_q = get_nnz(q_t); in sdpa_nested_preprocessing_backward() local