- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for _key_placeholder (0.13 sec)
-
tensorflow/compiler/jit/extract_outside_compilation_pass_test.cc
Graph *host_graph = host_fbody->graph; Node *key_placeholder = nullptr, *sequencer = nullptr; for (Node *n : host_graph->nodes()) { if (n->type_string() == "Placeholder" && absl::EndsWith(n->name(), "_key_placeholder")) { EXPECT_EQ(key_placeholder, nullptr); key_placeholder = n; } else if (HasNodeAttr(n->def(), "_xla_host_transfer_sequencer")) {
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
Graph* g = fbody->graph; // Find or create the key placeholder node. Node* key_placeholder = nullptr; for (Node* n : g->nodes()) { if (IsKeyPlaceholderNode(*n)) { key_placeholder = n; break; } } if (!key_placeholder) { TF_ASSIGN_OR_RETURN(key_placeholder, AddHostComputeKeyPlaceholder(xla_cluster_name, g)); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 12 06:33:33 UTC 2024 - 104.7K bytes - Viewed (0) -
tensorflow/compiler/jit/encapsulate_subgraphs_pass_test.cc
Node* KeyPlaceholderShape(const GraphDefBuilder::Options& opts) { return KnownShapeBase(DT_STRING, {2}, opts); } Node* KeyPlaceholder(const string& call_node, const GraphDefBuilder::Options& opts) { if (opts.HaveError()) return nullptr; NodeBuilder node_builder(absl::StrCat(call_node, "_key_placeholder"), "Placeholder", opts.op_registry()); TensorShapeProto shape;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 08:47:20 UTC 2024 - 113.3K bytes - Viewed (0)