- Sort Score
- Result 10 results
- Languages All
Results 1 - 7 of 7 for GetGlobalJitLevelForGraph (0.51 sec)
-
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) -
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) -
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) -
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) -
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) -
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) -
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)