namespace mlir {
namespace odml {
namespace {

// TODO - b/330337238: Surface these to other files when needed.
constexpr llvm::StringLiteral kCompositeNamespace = "odml.internal";
constexpr llvm::StringLiteral kGelu = "gelu";

std::string MakeCompositeName(llvm::StringRef op_name) {
  return (kCompositeNamespace + "." + op_name).str();
}

#define GEN_PASS_DEF_OUTLINECOMPOSITESPASS

          (variadic $inputs),
          ConstantStrAttr<StrAttr, "aten.gelu.default">, $attrs, $_, $_),
        (TFL_GeluOp $inputs, ConstBoolAttrFalse),
        [(IsStrCompositeAttribute<"approximate", "none"> $attrs)]>;

def LegalizeCompositeApproximateAtenGELU : Pat<
        (MHLO_CompositeOp:$composite
          (variadic $inputs),
          ConstantStrAttr<StrAttr, "aten.gelu.default">, $attrs, $_, $_),