#include "mlir/Pass/Pass.h"  // from @llvm-project

namespace mlir {
namespace tf_saved_model {

// Create a pass that removes function arguments that map to global tensors.
std::unique_ptr<Pass> CreateLowerGlobalsToMlProgramPass();

// Register this pass in the global registry of MLIR.
void RegisterLowerGlobalsToMlProgramPass();

}  // namespace tf_saved_model
}  // namespace mlir

      signalPassFailure();
    }
  }
};

std::unique_ptr<Pass> CreateLowerGlobalsToMlProgramPass() {
  return std::make_unique<LowerGlobalsToMlProgram>();
}

void RegisterLowerGlobalsToMlProgramPass() {
  registerPass(CreateLowerGlobalsToMlProgramPass);
}

}  // namespace tf_saved_model

  // calls.
  pm.addPass(mlir::TF::CreateTFFunctionalControlFlowToRegions());

  pm.addPass(mlir::tf_saved_model::CreateLowerVariableOpsToMlProgramPass());
  pm.addPass(mlir::tf_saved_model::CreateLowerGlobalsToMlProgramPass());
  pm.addPass(mlir::TF::CreateLocalizeVarHandlesPass());
  pm.addPass(mlir::tf_saved_model::CreateAddFunctionsForExportedNamesPass());
  pm.addPass(mlir::tf_saved_model::CreateStripSavedModuleMetadataPass());

std::unique_ptr<OperationPass<func::FuncOp>> CreateDedupBoundInputBindingPass();

// Create a pass that removes function arguments that map to global tensors.
std::unique_ptr<Pass> CreateLowerGlobalsToMlProgramPass();

// Create a pass that lowers variable read/write ops to ml_program ops.
std::unique_ptr<OperationPass<ModuleOp>>
CreateLowerVariableOpsToMlProgramPass();

        ...
      }

    to

      func @f() {
        %0 = ml_program.global_load @v
        ...
      }
  }];
  let constructor = "mlir::tf_saved_model::CreateLowerGlobalsToMlProgramPass()";
}

def LowerVariableOpsToMlProgramPass: Pass<"tf-saved-model-lower-variable-ops-to-mlprogram", "mlir::ModuleOp"> {
  let summary = "Lower tf.ReadVariable and tf.AssignVariable";

  let description = [{