EXPECT_TRUE(IsStablehloOp(*constant_op));
}

TEST_F(StablehloTypeUtilsTest, IsStablehloOpFailsWithArithOp) {
  const OwningOpRef<mlir::arith::ConstantOp> constant_op =
      builder_.create<mlir::arith::ConstantOp>(builder_.getUnknownLoc(),
                                               builder_.getI32IntegerAttr(0));
  EXPECT_FALSE(IsStablehloOp(*constant_op));
}

}  // namespace

#include "stablehlo/dialect/StablehloOps.h"  // from @stablehlo

namespace mlir::quant::stablehlo {

// Checks if an op is from StableHLO dialect.
inline bool IsStablehloOp(Operation* op) {
  return op->getDialect()->getNamespace() ==
         mlir::stablehlo::StablehloDialect::getDialectNamespace();
}

}  // namespace mlir::quant::stablehlo

    if (current_depth > 5) {
      all_descendants.clear();
      return false;
    }
    current_depth++;

    for (Operation* descendant : current_layer_descendants) {
      if (!IsStablehloOp(descendant) || !ops_to_add.contains(descendant)) {
        all_descendants.clear();
        return false;
      }
      for (Operation* next_descendant : descendant->getUsers()) {

}

bool IsInStableHloOpRegion(Operation* op) {
  if (op == nullptr) return false;
  auto parent_op = op->getParentOp();
  return parent_op != nullptr && stablehlo::IsStablehloOp(parent_op);
}

// Inserts the function to the symbol table of the module thread-safely.
StringAttr InsertToSymbolTable(Operation& module, Operation& function,
                               const StringRef func_name) {

    }
    // All encodings supported in StableHLO.
    return attr;
  }
};

// Returns true if the op_code belongs to a stablehlo operation.
bool IsStablehloOp(const tflite::OperatorCodeT &op_code);

// Returns the MLIR op name for the flatbuffer operator corresponding to
// `op_code`.
std::string GetMlirOpNameFromOpCode(const ::tflite::OperatorCodeT &op_code);

  }

  const char* option_name = tflite::EnumNamePadding(padding);
  return builder.getStringAttr(option_name);
}

}  // namespace

bool mlir::IsStablehloOp(const tflite::OperatorCodeT& op_code) {
  llvm::StringRef op_name(
      tflite::EnumNameBuiltinOperator(tflite::GetBuiltinCode(&op_code)));
  return op_name.starts_with("STABLEHLO_");
}