- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for pipelining (0.14 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/sparsecore/embedding_pipelining.cc
// Enable automated pipelining pass unless: // 1. The user disables it via flag, or // 2. The graph contains TF.Summary ops. Graphs like this typically only run // for a single step which doesn't work in pipelining. if (tensorflow::GetBuildXlaOpsPassFlags() ->tf_xla_disable_full_embedding_pipelining) { LOG(INFO) << "Embedding pipelining disabled via flag."; return false; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 92.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/g3doc/_includes/tf_passes.md
compiling to XLA. ### `-tf-einsum` _Transform Einsum to other TF Ops for the supported variants_ ### `-tf-embedding-pipelining` _Rewrite graph for embedding pipelining_ For architectures that support accelerated embedding lookups, this pass will rewrite the graph to use pipelining for better device utilization. ### `-tf-embedding-program-key` _Sets the program key for embedding ops._
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Aug 02 02:26:39 UTC 2023 - 96.4K bytes - Viewed (0)