// register extensions here, again.
    mlir::DialectRegistry registry;
    registry.insert<mlir::func::FuncDialect>();
    mlir::func::registerAllExtensions(registry);
    xla_call_module_context_.appendDialectRegistry(registry);

    auto l = tensorflow::XlaCallModuleLoader::Create(
        &xla_call_module_context_, op.getVersion(), op.getModule().str(),
        std::move(disabled_checks), std::move(platforms),

}

void LoadImporterDialects(mlir::MLIRContext& context) {
  // Load dialects involved in the conversion
  mlir::DialectRegistry registry;
  mlir::RegisterAllTensorFlowDialectsImpl(registry, false);
  context.appendDialectRegistry(registry);
  for (llvm::StringRef name : registry.getDialectNames())
    context.getOrLoadDialect(name);
}

absl::StatusOr<std::string> GetDenseTensorNameFromTensorInfo(
    const TensorInfo& tensor_info) {