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

struct MergeControlFlowPass
    : public impl::MergeControlFlowPassBase<MergeControlFlowPass> {
  void runOnOperation() override;
};

// Gets the IfRegion op and all of ops in the then and else branches.

    to 12 feature dimension, which has better performance on TPU.
  }];

  let constructor = "TFTPU::CreateTPUSpaceToDepthPass()";
}

def MergeControlFlowPass : Pass<"tf-merge-control-flow", "ModuleOp"> {
  let summary = "Merges IfRegion ops together with a common predicate.";

  let description = [{