hgSSKrSSKJr SSKJr SSjrS\R S\RS\R4SjrSSjr S\RS\RS\R4S jr SS jr S\RS\RS\R4S jrS\R4S jrS\R"4S jrS\R&4Sjrg)N)PackedSequencec6XU--U-nU(aX1S-- nU$)z`Calculate the total operations for an RNN cell given input size, hidden size, and optional bias.) input_size hidden_sizebias total_opss H/home/james-whalen/.local/lib/python3.13/site-packages/thop/rnn_hooks.py_count_rnn_cellr s)K78;FI 1_$ mxyc[URURUR5nUSR S5nX4-nU=R [ R"[U5/5- slg)z^Counts the total RNN cell operations based on input tensor, hidden size, bias, and batch size.rN) r rrr sizer torch DoubleTensorintrrrr batch_sizes r count_rnn_cellrU ammQVVDI11J IKK5%%s9~&677Kr cSnX-U-U-nU(aXAS-- nX4S-- nX1U-U-U-- nU(aX1S-- nX1- nX1S-- nU$)zdCounts the total operations for a GRU cell based on input size, hidden size, and bias configuration.rrrrrr r state_opss r _count_gru_cellrszI)[8;FI 1_$  QI *k9KGGI 1_$  Iq I r c[URURUR5nUSR S5nX4-nU=R [ R"[U5/5- slg)z\Calculates and updates the total operations for a GRU cell in a mini-batch during inference.rN) rrrr rr rrrrs r count_gru_cellr 3rr c^SnX-U-U-nU(aXAS-- nX4S-- nX1S-- nX1- nU$)zfCounts LSTM cell operations during inference based on input size, hidden size, and bias configuration.rrrrrs r _count_lstm_cellr#=sTI )[8;FI 1_$  QIq II r c[URURUR5nUSR S5nX4-nU=R [ R"[U5/5- slg)zZCounts and updates the total operations for an LSTM cell in a mini-batch during inference.rN) r#rrr rr rrrrs r count_lstm_cellr%TsU q}}affEI11J IKK5%%s9~&677Kr cJURnURnURnURn[ US[ 5(aB[ R"USR5nUSRRS5nObUR(a)USRS5nUSRS5nO(USRS5nUSRS5nSn UR(aU [XEU5S-- n OU [XEU5- n [US- 5H5n U UR(a[US-XS5S-O [XUU5- n M7 X-n X-n U=R[ R"[!U 5/5- slg)zWCalculate and update the total number of operations for each RNN cell in a given batch.rrN)r rr num_layers isinstancerrmax batch_sizesr batch_first bidirectionalr ranger rr rrrr rrr(r num_stepsr _s r count_rnnr2^\ 66DJ--KJ!A$''YYqt//0 aD$$))!, qTYYq\ aDIIaL qTYYq\ aDIIaL I_ZdCaGG _ZdCC :> " K!O[ ?! C 4@ #I IKK5%%s9~&677Kr cJURnURnURnURn[ US[ 5(aB[ R"USR5nUSRRS5nObUR(a)USRS5nUSRS5nO(USRS5nUSRS5nSn UR(aU [XEU5S-- n OU [XEU5- n [US- 5H5n U UR(a[US-XS5S-O [XUU5- n M7 X-n X-n U=R[ R"[!U 5/5- slg)zfCalculates total operations for a GRU layer, updating the model's operation count based on batch size.rr'rN)r rrr(r)rrr*r+rr,r-rr.r rrr/s r count_grur5r3r cJURnURnURnURn[ US[ 5(aB[ R"USR5nUSRRS5nObUR(a)USRS5nUSRS5nO(USRS5nUSRS5nSn UR(aU [XEU5S-- n OU [XEU5- n [US- 5H5n U UR(a[US-XS5S-O [XUU5- n M7 X-n X-n U=R[ R"[!U 5/5- slg)zgCalculate total operations for LSTM layers, including bidirectional, updating model's total operations.rr'rN)r rrr(r)rrr*r+rr,r-r#r.r rrr/s r count_lstmr7s^ 66DJ--KJ!A$''YYqt//0 aD$$))!, qTYYq\ aDIIaL qTYYq\ aDIIaL I%jtDqHH %jtDD :> " [1_k @1 D!+DA #I IKK5%%s9~&677Kr )T)rtorch.nnnntorch.nn.utils.rnnrr RNNCellTensorrrGRUCellr r#LSTMCellr%RNNr2GRUr5LSTMr7rr r rBs -8bjj8U\\8ell808bjj8U\\8ell8.8r{{8u||8 8"8"8J"8"8J"8"''"8r