torch.utils.flop_counter.sdpa_flop

torch.utils.flop_counter.sdpa_flop(query_shape, key_shape, value_shape, *args, out_shape=None, **kwargs)

Count flops for self-attention.

Return type:

int