Facilitate graph pattern matching

[ricardoV94]

This PR provides a POC for #1455

• It adds two variable properties: owner_op and owner_op_and_inputs.
  • This allows us to avoid the verbose if x.owner.op is not None and isinstance(x.owner.op, ...) in favor of isinstance(x.owner_op, ...)
  • The second one is more niche, but allows to both match an Op and structurally unpack the inputs in case of a match in a single case statement. This makes nested patterns easier to write. Eg. case (Blocwise(Dot()), a, b) for a single match x.owner_op_and_inputs.
• It adds match_args so common parametrized Ops (Elemwise, Blockwise) and Constants are easier to pattern match without kwargs
• It adds a much needed is_matrix_transpose and is_left_expanded_matrix_transpose property to DimShuffle, for easier pattern matching

I then refactored the existing linalg rewrites as a POC. I went a bit overboard and tried to standardize names, and fuse rewrites that were pretty simple to combine.

CC @tanish1729 who did many of these rewrites, see if it feels nicer?

[ricardoV94]

This is new, we were not checking for non standard allocdiagonal, not along the last dims!