Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 3 of 3 for tf_xla_ops_to_cluster (0.16 sec)

  1. tensorflow/compiler/jit/flags.cc

               &mark_for_compilation_flags->tf_xla_max_cluster_size,
               "Maximum number of operators in an XLA compilation."),
          Flag(
              "tf_xla_ops_to_cluster",
              &mark_for_compilation_flags->tf_xla_ops_to_cluster,
              "(experimental) "
              "Limit the operations clustered by XLA to these operations. "
              "If multiple, separate them with commas. Shortcuts: "
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Apr 17 18:52:57 UTC 2024
    - 24.5K bytes
    - Viewed (0)
  2. tensorflow/compiler/jit/flags.h

      int32 tf_xla_min_cluster_size;
    
      // Maximum number of operators in an XLA compilation.
      int32 tf_xla_max_cluster_size;
    
      // If non-empty, limit XLA clustering to the following TF operations.
      string tf_xla_ops_to_cluster;
    
      // If non-empty, remove following operations from XLA clustering excludelist.
      string tf_xla_cluster_exclude_ops;
    
      // Dump graphs during XLA compilation.
      bool tf_xla_clustering_debug;
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Apr 17 18:52:57 UTC 2024
    - 14.5K bytes
    - Viewed (0)
  3. tensorflow/compiler/jit/mark_for_compilation_pass.cc

          tensorflow::GetAllowlistTable();
      MarkForCompilationPassFlags* flags = GetMarkForCompilationPassFlags();
      absl::flat_hash_set<string> allowlist;
    
      for (auto s : absl::StrSplit(flags->tf_xla_ops_to_cluster, ',')) {
        if (s == "FUSIBLE") {
          for (auto pair : *allowlist_table) {
            allowlist.insert(pair.second.begin(), pair.second.end());
          }
        } else if (allowlist_table->contains(s)) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Feb 21 12:19:41 UTC 2024
    - 85.3K bytes
    - Viewed (0)
Back to top