_create_sparse_block_from_block_mask (reference) in projects: aosp_15_r20

Searched refs:_create_sparse_block_from_block_mask (Results 1 – 1 of 1) sorted by relevance

/aosp_15_r20/external/pytorch/torch/nn/attention/
H A D	flex_attention.py	`683 def _create_sparse_block_from_block_mask( function 853 block_mask = _create_sparse_block_from_block_mask(`