- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for _key_placeholder (0.37 sec)
-
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)