namespace cutil = TF::collection_ops_util;

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

struct StackOpsDecompositionPass
    : public impl::StackOpsDecompositionPassBase<StackOpsDecompositionPass> {
  void runOnOperation() final;
};

// Returns the type of the local variable for the stack size.
Type GetSizeVarType(OpBuilder builder) {

            tf_executor.yield
          }
          tf_executor.fetch %control : !tf_executor.control
        }
        return
      }
    ```
  }];
}

def StackOpsDecompositionPass : Pass<"tf-stack-ops-decomposition", "ModuleOp"> {
  let summary = "Decompose stack operations into local variable operations. Needs "
           "static shapes.";