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

struct ResourceOpLiftingForMainFunctionPass
    : public impl::ResourceOpLiftingForMainFunctionPassBase<
          ResourceOpLiftingForMainFunctionPass> {
  void runOnOperation() override;
};

void ResourceOpLiftingForMainFunctionPass::runOnOperation() {
  ModuleOp module = getOperation();