- Sort Score
- Result 10 results
- Languages All
Results 1 - 5 of 5 for kXlaConnectedToXlaComputationAttrName (0.45 sec)
-
tensorflow/compiler/jit/encapsulate_util.cc
e->src()->name())); } } else if (src_outside_compilation && !dst_outside_compilation) { // Case 1b: outside compilation to its XLA computation control edge. ReplaceAttr(e->src(), kXlaConnectedToXlaComputationAttrName, true); } else if (!src_outside_compilation && dst_outside_compilation) { // Case 1b: XLA computation to outside compilation in it control edge.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 12 06:33:33 UTC 2024 - 15.5K bytes - Viewed (0) -
tensorflow/compiler/jit/extract_outside_compilation_pass.h
// 4. Mark all nodes except key placeholder with attr `xla_cluster_attr_name` // and `outside_compilation_attr_name`; // 5. For nodes marked with attr kXlaConnectedToXlaComputationAttrName, add a // control edge from the node to XlaSendFromHost; for nodes marked with attr // kXlaConnectedFromXlaComputationAttrName, add a control edge from // XlaRecvAtHost node to the node;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 06 19:12:29 UTC 2023 - 5.3K bytes - Viewed (0) -
tensorflow/compiler/jit/encapsulate_util.h
// Attribute indicating that some ops in this node's XLA computation has control // dependency on this node. Attribute value will always be "true". extern const char kXlaConnectedToXlaComputationAttrName[]; // Attribute indicating that this node has control dependency on some ops in // this node's XLA computation. Attribute value will always be "true". extern const char kXlaConnectedFromXlaComputationAttrName[];
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 06:59:07 UTC 2024 - 7.4K bytes - Viewed (0) -
tensorflow/compiler/jit/extract_outside_compilation_pass_test.cc
TF_CHECK_OK(s.ToGraph(g.get())); auto node_name_image = g->BuildNodeNameIndex(); Node *add_node = node_name_image["add"]; EXPECT_NE(add_node, nullptr); add_node->AddAttr(kXlaConnectedToXlaComputationAttrName, "cluster"); add_node->AddAttr(kXlaConnectedFromXlaComputationAttrName, "cluster"); RewriteOutsideCompilationSubgraphFn rewrite_fn("_xla", "_oc", "cluster", ""); std::vector<OutputTensor> arg_source_tensors;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 06 19:12:29 UTC 2023 - 41K bytes - Viewed (0) -
tensorflow/compiler/jit/extract_outside_compilation_pass.cc
// control edges. for (Node* n : (*graph)->nodes()) { if (HasNodeAttr(n->def(), kXlaConnectedToXlaComputationAttrName)) { (*graph)->AddControlEdge(n, send_from_host_node); n->ClearAttr(kXlaConnectedToXlaComputationAttrName); } if (HasNodeAttr(n->def(), kXlaConnectedFromXlaComputationAttrName)) { (*graph)->AddControlEdge(recv_at_host_node, n);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 12 06:33:33 UTC 2024 - 104.7K bytes - Viewed (0)