Searched defs:_validate_sdpa_input (Results 1 – 2 of 2) sorted by relevance
| /external/pytorch/torch/nn/attention/ | ||
| D | _utils.py | 42 def _validate_sdpa_input( function |
| /external/pytorch/torch/nested/_internal/ | ||
| D | sdpa.py | 23 def _validate_sdpa_input( function |