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

class MaterializePassthroughOpPass
    : public impl::MaterializePassthroughOpBase<MaterializePassthroughOpPass> {
 public:
  void runOnOperation() override;
};

void MaterializePassthroughOpPass::runOnOperation() {
  getOperation().walk([](TF::MlirPassthroughOp op) {
    std::string module_string(op.getMlirModule());