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

class InsertCalibrationStatisticsSaverPass
    : public impl::InsertCalibrationStatisticsSaverPassBase<
          InsertCalibrationStatisticsSaverPass> {
 public:
  using impl::InsertCalibrationStatisticsSaverPassBase<
      InsertCalibrationStatisticsSaverPass>::
      InsertCalibrationStatisticsSaverPassBase;

 private:

    make much sense or sharding custom calls are incompatible, e.g. on-device
    targets.
  }];
  let dependentDialects = ["mlir::stablehlo::StablehloDialect"];
}

def InsertCalibrationStatisticsSaverPass : Pass<"stablehlo-insert-calibration-statistics-saver", "ModuleOp"> {
  let summary = "Inserts `CalibrationStatisticsSaver` op to collect and save calibration statistics.";
  let description = [{