- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for sparsecore_passes (0.12 sec)
-
tensorflow/compiler/mlir/tf2xla/internal/clustering_bridge_passes.cc
#include "mlir/Transforms/Passes.h" // from @llvm-project #include "tensorflow/compiler/jit/flags.h" #include "tensorflow/compiler/mlir/tensorflow/transforms/passes.h" #include "tensorflow/compiler/mlir/tensorflow/transforms/sparsecore/sparsecore_passes.h" #include "tensorflow/compiler/mlir/tf2xla/internal/passes/clustering_passes.h" namespace tensorflow { namespace tf2xla { namespace internal { using mlir::OpPassManager; using mlir::func::FuncOp;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 16:09:14 UTC 2024 - 11.2K bytes - Viewed (1) -
tensorflow/compiler/mlir/tensorflow/transforms/sparsecore/embedding_program_key.cc
constexpr char kMiniBatchSplitsAttr[] = "mini_batch_splits"; constexpr char kMiniBatchCsrAttr[] = "mini_batch_in_csr"; #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`.
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/embedding_sequencing.cc
#include "tensorflow/compiler/mlir/tensorflow/utils/attribute_utils.h" #define GEN_PASS_DEF_EMBEDDINGSEQUENCINGPASS #include "tensorflow/compiler/mlir/tensorflow/transforms/sparsecore/sparsecore_passes.h.inc" static constexpr char kEmbeddingPipelining[] = "_embedding_pipelining"; static constexpr char kEmbeddingForward[] = "forward"; static constexpr char kEmbeddingBackward[] = "backward";
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 39.4K bytes - Viewed (0)