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