- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for EmbeddingPipeliningPass (0.69 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/sparsecore/sparsecore_passes.td
limitations under the License. ==============================================================================*/ include "mlir/Pass/PassBase.td" def EmbeddingPipeliningPass : Pass<"tf-embedding-pipelining", "mlir::ModuleOp"> { let summary = "Rewrite graph for embedding pipelining"; let constructor = "TFDevice::CreateEmbeddingPipeliningPass()"; 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) -
tensorflow/compiler/mlir/tf2xla/internal/clustering_bridge_passes.cc
pm.addNestedPass<FuncOp>(mlir::TF::CreateDecomposeReduceDatasetPass()); // Only one of EmbeddingSequencing and EmbeddingPipelining will actually // run and the logic is in EmbeddingPipeliningPass. If the pipelining pass // runs, embedding attributes are stripped and the sequencing pass will have // no effect. If the pipelining pass doesn't run, embedding attributes are
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 16:09:14 UTC 2024 - 11.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/sparsecore/embedding_pipelining.cc
if (mlir::isa<TF::ResourceType>(tensor_type.getElementType())) { return true; } } return false; } struct EmbeddingPipeliningPass : public ::impl::EmbeddingPipeliningPassBase<EmbeddingPipeliningPass> { void getDependentDialects(mlir::DialectRegistry& registry) const override { registry.insert<TF::TensorFlowDialect>(); } void runOnOperation() override; };
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 92.9K bytes - Viewed (0)