#include "tensorflow/compiler/mlir/quantization/stablehlo/passes/passes.h.inc"

namespace {

class OptimizeGraphPass
    : public impl::OptimizeGraphPassBase<OptimizeGraphPass> {
 public:
  MLIR_DEFINE_EXPLICIT_INTERNAL_INLINE_TYPE_ID(OptimizeGraphPass)

  explicit OptimizeGraphPass() = default;

 private:
  void runOnOperation() override;
};

  let dependentDialects = [
    "TF::TensorFlowDialect",
    "mlir::quant::QuantizationDialect",
    "mlir::shape::ShapeDialect",
    "mlir::stablehlo::StablehloDialect",
  ];
}

def OptimizeGraphPass : Pass<"optimize-graph", "ModuleOp"> {
  let summary = "Optimize the sub-optimal patterns after quantization.";
  let dependentDialects = ["mlir::stablehlo::StablehloDialect",];
}