#include "tensorflow/compiler/mlir/tensorflow/transforms/tf_passes.h.inc"

class PromoteVarHandlesToArgsPass
    : public impl::PromoteVarHandlesToArgsPassBase<
          PromoteVarHandlesToArgsPass> {
 public:
  void runOnOperation() override;
};

void PromoteVarHandlesToArgsPass::runOnOperation() {
  ModuleOp module = getOperation();
  MLIRContext* context = module.getContext();