Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 7 of 7 for GetGlobalJitLevelForGraph (0.43 sec)

  1. tensorflow/compiler/jit/report_clustering_info_pass.cc

        const GraphOptimizationPassOptions& options) {
      XlaAutoClusteringActivity activity;
      *activity.mutable_summary() = GetXlaAutoClusteringSummary(**options.graph);
      activity.set_global_jit_level(GetGlobalJitLevelForGraph(options));
      activity.set_cpu_global_jit_enabled(
          GetMarkForCompilationPassFlags()->tf_xla_cpu_global_jit);
      return BroadcastXlaActivity(std::move(activity));
    }
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Sat Jun 15 00:14:19 UTC 2019
    - 1.3K bytes
    - Viewed (0)
  2. tensorflow/compiler/jit/xla_activity.proto

    }
    
    // Listeners listening for auto clustering events get messages of this type.
    //
    // Next ID: 4
    message XlaAutoClusteringActivity {
      // The value of GlobalJitLevel, as determined by `GetGlobalJitLevelForGraph`.
      // This determines if global auto-clustering is enabled.
      OptimizerOptions.GlobalJitLevel global_jit_level = 1;
    
      // Whether --tf_xla_cpu_global_jit is enabled in TF_XLA_FLAGS.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Mar 15 03:11:33 UTC 2022
    - 3.6K bytes
    - Viewed (0)
  3. tensorflow/compiler/jit/xla_cluster_util.h

    bool HasResourceInputOrOutput(const Node& node);
    
    // Determines the global jit level based on GraphOptimizationPassOptions,
    // --tf_xla_auto_jit and whether the graph is a single GPU graph.
    OptimizerOptions::GlobalJitLevel GetGlobalJitLevelForGraph(
        const GraphOptimizationPassOptions& options);
    
    // Returns true if `g` is a single-GPU graph.  A single-GPU graph uses exactly
    // one GPU (and any number of CPUs).
    bool IsSingleGpuGraph(const Graph& g);
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Feb 21 09:53:30 UTC 2024
    - 4.6K bytes
    - Viewed (0)
  4. tensorflow/compiler/jit/xla_cluster_util.cc

        VLOG(4) << "GetGlobalJitLevelForGraph returning "
                << xla_global_jit_level.single_gpu;
        return xla_global_jit_level.single_gpu;
      }
      OptimizerOptions::GlobalJitLevel result =
          IsSingleGpuGraph(**options.graph) ? xla_global_jit_level.single_gpu
                                            : xla_global_jit_level.general;
      VLOG(4) << "GetGlobalJitLevelForGraph returning " << result;
      return result;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Feb 29 08:39:39 UTC 2024
    - 21.3K bytes
    - Viewed (0)
  5. tensorflow/compiler/jit/clone_constants_for_better_clustering.cc

      }
      return absl::OkStatus();
    }
    
    Status CloneConstantsForBetterClusteringPass::Run(
        const GraphOptimizationPassOptions& options) {
      if (GetGlobalJitLevelForGraph(options) == OptimizerOptions::OFF) {
        return absl::OkStatus();
      }
    
      Graph* g = options.graph->get();
    
      if (VLOG_IS_ON(1)) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Mar 12 06:33:33 UTC 2024
    - 7.3K bytes
    - Viewed (0)
  6. tensorflow/compiler/jit/cluster_scoping_pass.cc

    }
    }  // namespace
    
    Status ClusterScopingPass::Run(const GraphOptimizationPassOptions& options) {
      Graph* graph = options.graph->get();
    
      return ClusterScopingPassImpl{graph, GetGlobalJitLevelForGraph(options)}
          .Run();
    }
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Feb 22 08:47:20 UTC 2024
    - 5.7K bytes
    - Viewed (0)
  7. tensorflow/compiler/jit/mark_for_compilation_pass.cc

          debug_options,
          graph,
          flib_def,
          options.session_options != nullptr ? options.session_options->env
                                             : Env::Default(),
          GetGlobalJitLevelForGraph(options),
          options.session_options->config.graph_options()
              .optimizer_options()
              .cpu_global_jit(),
          /*cluster_name_prefix=*/options.session_options != nullptr
    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