A counter for the `thop` package to count the operations in an attention operation. Meant to be used like: macs, params = thop.profile( model, inputs=(inputs, timestamps), custom_ops={QKVAttention: QKVAttention.count_flops}, )
(model, _x, y)
source not stored for this graph (policy: none)
no outgoing calls
no test coverage detected