torch.utils.flop\_counter.sdpa\_flop\_count =========================================== .. currentmodule:: torch.utils.flop_counter .. autofunction:: sdpa_flop_count