1*da0073e9SAndroid Build Coastguard Worker.. role:: hidden 2*da0073e9SAndroid Build Coastguard Worker :class: hidden-section 3*da0073e9SAndroid Build Coastguard Worker 4*da0073e9SAndroid Build Coastguard Workertorch.nn.attention 5*da0073e9SAndroid Build Coastguard Worker================== 6*da0073e9SAndroid Build Coastguard Worker 7*da0073e9SAndroid Build Coastguard Worker.. automodule:: torch.nn.attention 8*da0073e9SAndroid Build Coastguard Worker 9*da0073e9SAndroid Build Coastguard WorkerUtils 10*da0073e9SAndroid Build Coastguard Worker------------------- 11*da0073e9SAndroid Build Coastguard Worker.. autosummary:: 12*da0073e9SAndroid Build Coastguard Worker :toctree: generated 13*da0073e9SAndroid Build Coastguard Worker :nosignatures: 14*da0073e9SAndroid Build Coastguard Worker 15*da0073e9SAndroid Build Coastguard Worker sdpa_kernel 16*da0073e9SAndroid Build Coastguard Worker SDPBackend 17*da0073e9SAndroid Build Coastguard Worker 18*da0073e9SAndroid Build Coastguard WorkerSubmodules 19*da0073e9SAndroid Build Coastguard Worker---------- 20*da0073e9SAndroid Build Coastguard Worker.. autosummary:: 21*da0073e9SAndroid Build Coastguard Worker :nosignatures: 22*da0073e9SAndroid Build Coastguard Worker 23*da0073e9SAndroid Build Coastguard Worker flex_attention 24*da0073e9SAndroid Build Coastguard Worker bias 25*da0073e9SAndroid Build Coastguard Worker 26*da0073e9SAndroid Build Coastguard Worker.. toctree:: 27*da0073e9SAndroid Build Coastguard Worker :hidden: 28*da0073e9SAndroid Build Coastguard Worker 29*da0073e9SAndroid Build Coastguard Worker nn.attention.flex_attention 30*da0073e9SAndroid Build Coastguard Worker nn.attention.bias