}
  // unary einsum op is only supported to the case where the operation can be
  // replaced using reduce_sum and/or transpose
  if (const auto dnums_or =
          GetEinsumDimensionNumbersUnary(op.getEquation(), lhs)) {
    return rewriteToReduceSumAndTranspose(op, dnums_or.value(), rewriter);
  }

  return rewriter.notifyMatchFailure(op, "unsupported einsum lowering");
}

#define GEN_PASS_DEF_TRANSFORMEINSUMPASS