void runOnOperation() override;
};

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

struct LaunchOutliningPass
    : public impl::LaunchOutliningPassBase<LaunchOutliningPass> {
  void runOnOperation() override;
};

void ReplaceClusterReturnWithReturn(tf_device::ReturnOp cluster_return_op,
                                    OpBuilder* builder) {

    dialect to the SCF dialect.
  }];

  let dependentDialects = ["scf::SCFDialect", "tensor::TensorDialect"];

  let constructor = "TF::createConvertTfControlFlowToScfPass()";
}

def LaunchOutliningPass : Pass<"tf-device-launch-outlining", "ModuleOp"> {
  let summary = "Outlines regions of tf_device.launch operations";

  let description = [{
    This pass outlines the body of a `tf_device.launch` into a function and