- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for kXlaCompileAttr (0.15 sec)
-
tensorflow/compiler/jit/mark_for_compilation_pass_test.cc
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 14 10:11:10 UTC 2024 - 79.6K bytes - Viewed (0) -
tensorflow/compiler/jit/mark_for_compilation_pass.cc
bool compile = false; Status status = GetNodeAttr(node->attrs(), kXlaCompileAttr, &compile); if (status.ok()) { if (!compile) { VLOG(2) << "Rejecting " << node->name() << ": kXlaCompileAttr(" << kXlaCompileAttr << ") is false."; } return !compile; } status = flib_def_->GetAttr(*node, kXlaCompileAttr, &compile); if (status.ok()) { if (!compile) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 21 12:19:41 UTC 2024 - 85.3K bytes - Viewed (0) -
tensorflow/compiler/jit/compilability_check_util.cc
static bool NodeCanTriggerXlaCompilation(const NodeDef& node) { return node.attr().find(kXlaClusterIdAttr) != node.attr().end() || HasBoolAttr(node, kXlaMustCompileAttr) || HasBoolAttr(node, kXlaCompileAttr) || HasBoolAttr(node, kXlaScopeAttr) || HasBoolAttr(node, kXlaInternalScopeAttr) || ops_triggering_xla_compilation->count(node.op()); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 12 06:33:33 UTC 2024 - 30.3K bytes - Viewed (0)