Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/nn/attention/
H A Dflex_attention.py176 def _dense_to_ordered(dense_mask) -> Tuple: argument
/aosp_15_r20/external/pytorch/aten/src/ATen/native/sparse/
H A DSparseCsrTensorMath.cpp353 auto dense_mask = at::_sparse_compressed_tensor_unsafe( in sparse_mask_sparse_compressed() local