As per title, baddbmm is not supported and is quite common
Can submit PR with below unless any issues/concerns...
def baddbmm_flop_jit(inputs: List[Any], outputs: List[Any]) -> Number:
"""
Count flops for fully connected layers.
"""
input_shapes = [get_shape(v) for v in inputs[1:3]]
n, c, t = input_shapes[0]
d = input_shapes[-1][-1]
flop = n * c * t * d
return flop
As per title, baddbmm is not supported and is quite common
Can submit PR with below unless any issues/concerns...