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

class VerifyNoOutsideCompilationMarkersPass
    : public impl::VerifyNoOutsideCompilationMarkersPassBase<
          VerifyNoOutsideCompilationMarkersPass> {
 public:
  void runOnOperation() override;
};

bool IsLaunchOp(Operation& op) {
  return dyn_cast<tf_device::LaunchOp>(op) != nullptr;
}

      }) {device = "CPU:0"} : () -> tensor<i1>
      return %1
    }
    ```
  }];

  let constructor = "TFDevice::CreateDeviceAttributeToLaunchPass()";
}

def VerifyNoOutsideCompilationMarkersPass : Pass<"verify-no-outside-compilation-markers", "mlir::func::FuncOp"> {