allow more ops to einsum