Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/nn/attention/
H A Dbias.py19 _postprocess_flash_output,
243 return _postprocess_flash_output(out, og_head_size)
H A D_utils.py17 def _postprocess_flash_output(inpt_tensor: torch.Tensor, og_size: int) -> torch.Tensor: function