- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for EmbeddingProgramKeyPass (0.15 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/sparsecore/embedding_program_key.cc
#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();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 17.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/sparsecore/sparsecore_passes.td
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 = [{
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Mar 28 23:42:09 UTC 2024 - 3.9K bytes - Viewed (0)