dialect.addOperations<
#define GET_OP_LIST
#include "tensorflow/compiler/mlir/quantization/tensorflow/passes/tf_quant_ops.cc.inc"
      >();
}

void RegisterOps() {
  TF_DIALECT_REGISTER_ADDITIONAL_OPERATIONS(RegisterOpsHook);
}

static auto kRegistration = (RegisterOps(), true);

}  // namespace quant

#define GET_OP_CLASSES
#include "tensorflow/compiler/mlir/quantization/tensorflow/passes/tf_quant_ops.h.inc"

namespace mlir {
namespace quant {

// Function to register TensorFlow Uniform Quantized ops.
void RegisterOps();

}  // namespace quant
}  // namespace mlir

  stablehlo::registerAllDialects(registry);
  registry.insert<mlir::func::FuncDialect, mlir::tf_type::TFTypeDialect,
                  mlir::quant::QuantizationDialect>();
  mlir::quant::RegisterOps();

  MLIRContext context(registry);
  context.loadAllAvailableDialects();

  llvm::SourceMgr source_mgr;
  SourceMgrDiagnosticHandler sourceMgrHandler(source_mgr, &context);