#define GEN_PASS_DEF_EMBEDDINGPROGRAMKEYPASS
#include "tensorflow/compiler/mlir/tensorflow/transforms/sparsecore/sparsecore_passes.h.inc"

struct EmbeddingProgramKeyPass
    : public impl::EmbeddingProgramKeyPassBase<EmbeddingProgramKeyPass> {
  void runOnOperation() override;
};

// Checks if `op` is nested in `block`.
bool OpInBlock(Operation* op, Block* block) {
  Block* op_block = op->getBlock();

    let description = [{
    This is a strictly sequential and formally correct fallback option for the
    embedding pipelining pass intended for debugging during pipelining
    development.
  }];
}

def EmbeddingProgramKeyPass : Pass<"tf-embedding-program-key", "mlir::func::FuncOp"> {
  let summary = "Sets the program key for embedding ops.";
  let constructor = "TFDevice::CreateEmbeddingProgramKeyPass()";
    let description = [{