Home
last modified time | relevance | path

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

/external/pytorch/torch/nn/attention/
Dflex_attention.py611 Q_BLOCK_SIZE=_DEFAULT_SPARSE_BLOCK_SIZE, argument
672 Q_BLOCK_SIZE=_DEFAULT_SPARSE_BLOCK_SIZE, argument