- Sort Score
- Result 10 results
- Languages All
Results 1 - 1 of 1 for GatherNodesWithType (0.13 sec)
-
tensorflow/compiler/jit/extract_outside_compilation_pass.cc
bool IsKeyPlaceholderNode(const Node& n) { return n.type_string() == "Placeholder" && absl::EndsWith(n.name(), "_key_placeholder"); } // Returns nodes with given type. std::vector<Node*> GatherNodesWithType(const Graph& g, const string& type) { std::vector<Node*> result; for (Node* n : g.nodes()) { if (n->type_string() == type) { result.push_back(n); } } return result; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 12 06:33:33 UTC 2024 - 104.7K bytes - Viewed (0)