Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for pipelining (0.1 sec)

  1. 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)
  2. 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)
Back to top