#
n GLSL.std.450
main * : B J ` g @ �
GL_GOOGLE_cpp_style_line_directive GL_GOOGLE_include_directive main idx gl_GlobalInvocationID Pc total_elements num_heads head_idx & b_val ( B ( b_in * / beta_val 6 a_val 8 A 8 a_in : > a_log_val @ ALog @ a_log B F dt_bias_val H DtBias H dt_bias J N sp_input R softplus_val W g_val ^ BetaOut ^ beta_out ` e GOut e g_out g G H # H # G G ' H ( H ( # G ( G * " G * ! G 7 H 8 H 8 # G 8 G : " G : ! G ? H @ H @ # G @ G B " G B ! G G H H H H # G H G J " G J ! G ] H ^ H ^ # G ^ G ` " G ` ! G d H e H e # G e G g " G g ! G m !
;
+
; + + $ % $ ' $ ( ' ) ( ; ) * , $ + $ 0 �? 7 $ 8 7 9 8 ; 9 : ? $ @ ? A @ ; A B G $ H G I H ; I J ] $ ^ ] _ ^ ; _ ` d $ e d f e ; f g + k @ + l , m k l l 6 � ; ; ; % & ; % / ; % 6 ; % > ; % F ; % N ; % R ; % W A
= > = A = � � � � � � = A ! = " ! � # " > # = + A , - * + = $ . - > & . = $ 1 & $ 2 1 $ 3 2 � $ 4 0 3 � $ 5 0 4 > / 5 = ; A , < : ; = $ = < > 6 = = C A , D B C = $ E D > > E = K A , L J K = $ M L > F M = $ O 6 = $ P F � $ Q O P > N Q = $ S N $ T S � $ U 0 T $ V U > R V = $ X > $ Y X $ Z Y = $ [ R � $ \ Z [ > W \ = a = $ b / A , c ` a > c b = h = $ i W A , j g h > j i � 8