Searched defs:MultiheadAttentionForwardFuncOptions (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/csrc/api/include/torch/nn/options/ | ||
| H A D | activation.h | 655 struct TORCH_API MultiheadAttentionForwardFuncOptions { struct |
| /aosp_15_r20/external/pytorch/torch/csrc/api/src/nn/options/ | ||
| H A D | activation.cpp | 40 MultiheadAttentionForwardFuncOptions::MultiheadAttentionForwardFuncOptions( in MultiheadAttentionForwardFuncOptions() function in torch::nn::functional::MultiheadAttentionForwardFuncOptions |