- Sort Score
- Result 10 results
- Languages All
Results 1 - 4 of 4 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.h
std::vector<int>* constant_arg_indices, std::vector<int>* resource_arg_indices); // Given a NodeDef `node_def` returns true iff `node_def` has kXlaCompileAttr // set. bool CanCreateXlaKernel(const NodeDef& node_def); // Returns memory types for the input. // `constant_arg_indices` and `resource_arg_indices` are sorted arrays of
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 06 19:12:29 UTC 2023 - 14.9K 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)