Home
last modified time | relevance | path

Searched refs:NodeMap (Results 1 – 25 of 44) sorted by relevance

12

/external/tensorflow/tensorflow/core/grappler/
Dutils.cc75 NodeMap::NodeMap(GraphDef* graph) { in NodeMap() function in tensorflow::grappler::NodeMap
93 void NodeMap::RemoveNode(const string& name) { in RemoveNode()
98 NodeDef* NodeMap::GetNode(const string& name) const { in GetNode()
108 bool NodeMap::NodeExists(const string& name) const { in NodeExists()
113 const std::set<NodeDef*>& NodeMap::GetOutputs(const string& node_name) const { in GetOutputs()
121 void NodeMap::AddNode(const string& node_name, NodeDef* node) { in AddNode()
127 void NodeMap::AddOutput(const string& node_name, const string& output_name) { in AddOutput()
134 void NodeMap::RemoveOutput(const string& node_name, const string& output_name) { in RemoveOutput()
138 void NodeMap::UpdateInput(const string& node_name, const string& old_input_name, in UpdateInput()
144 void NodeMap::RemoveInputs(const string& node_name) { in RemoveInputs()
[all …]
Dutils.h45 class NodeMap {
49 explicit NodeMap(GraphDef* graph);
219 bool HasRegularOutputs(const NodeDef& node, const NodeMap& node_map);
222 bool HasControlOutputs(const NodeDef& node, const NodeMap& node_map);
231 int NumControlOutputs(const NodeDef& node, const NodeMap& node_map);
234 int NumNonControlOutputs(const NodeDef& node, const NodeMap& node_map);
238 int NumNonControlDataOutputs(const NodeDef& node, const NodeMap& node_map);
262 NodeDef* GetTailOfChain(const NodeDef& source, const NodeMap& node_map,
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dlayout_optimizer_test.cc234 NodeMap node_map(&output); in TEST_F()
269 NodeMap node_map(&output); in TEST_F()
290 NodeMap node_map(&output); in TEST_F()
303 NodeMap node_map(&output); in TEST_F()
316 NodeMap node_map(&output); in TEST_F()
329 NodeMap node_map(&output); in TEST_F()
342 NodeMap node_map(&output); in TEST_F()
355 NodeMap node_map(&output); in TEST_F()
370 NodeMap node_map(&output); in TEST_F()
399 NodeMap node_map_original(&item.graph); in TEST_F()
[all …]
Dscoped_allocator_optimizer.h33 class NodeMap; variable
76 NodeMap* node_map() { return node_map_.get(); } in node_map()
120 std::unique_ptr<NodeMap> node_map_;
Dgraph_optimizer_stage_test.cc92 NodeMap node_map(&graph); in TEST_F()
121 NodeMap node_map(&graph); in TEST_F()
154 NodeMap node_map(&item.graph); in TEST_F()
196 NodeMap node_map(&item.graph); in TEST_F()
Dscoped_allocator_optimizer.cc146 void RemoveNode(NodeDef* nd, GraphDef* graph, NodeMap* node_map) { in RemoveNode()
164 NodeDef* to_node, NodeMap* node_map) { in RemoveEdge()
212 NodeMap* node_map, const DataType& dtype, in MaybeRewriteInput()
254 NodeMap* node_map, const std::vector<NodeDef*>& ops, in GetInputs()
308 Status GetDataInputs(GraphDef* graph, NodeMap* node_map, NodeDef* op, in GetDataInputs()
405 NodeMap* node_map) { in ClearInternalControlInputs()
428 NodeMap* node_map, const std::vector<NodeDef*>& ops, in AnalyzeInputs()
463 GraphDef* graph, NodeMap* node_map, in TransitiveFanoutWithinFrame()
490 ScopedAllocatorOptimizer* sa_opti, GraphDef* graph, NodeMap* node_map, in ConstructScopedAllocatorNode()
570 Status BuildSAConcatNode(GraphDef* graph, NodeMap* node_map, in BuildSAConcatNode()
[all …]
Dscoped_allocator_optimizer_test.cc291 void GetNode(NodeMap* node_map, const string& node_name, NodeDef** node_def) { in GetNode()
298 NodeDef* ValidateSAControlInput(GraphDef* graph, NodeMap* node_map, in ValidateSAControlInput()
317 int NumControlInputs(NodeMap* node_map, const string& node_name) { in NumControlInputs()
347 NodeMap node_map(&optimized_graph); in TEST_F()
465 NodeMap node_map(&optimized_graph); in TEST_F()
498 NodeMap node_map(&optimized_graph); in TEST_F()
Dmodel_pruner.cc200 const NodeMap& node_map, const std::vector<string>& terminal_nodes, in IdentityNTerminalPorts()
273 GraphDef* graph, NodeMap* node_map) { in NewIdentityFromIdentityN()
298 NodeMap* node_map) { in RewriteIdentityNAndInputsOutputs()
391 NodeMap node_map(graph); in SplitIdentityNInputs()
Dmemory_optimizer_test.cc55 NodeMap pre_transform_node_map(&item.graph); in TEST_F()
64 NodeMap post_transform_node_map(&output); in TEST_F()
95 NodeMap pre_transform_node_map(&item.graph); in TEST_F()
118 NodeMap pre_transform_node_map(&item.graph); in TEST_F()
130 NodeMap post_transform_node_map(&output); in TEST_F()
158 NodeMap pre_transform_node_map(&item.graph); in TEST_F()
170 NodeMap post_transform_node_map(&first_pass_output); in TEST_F()
Dloop_optimizer.h67 const NodeMap& node_map,
Dgraph_optimizer_stage.h51 GraphProperties* graph_properties, NodeMap* node_map, in GraphOptimizerContext()
64 NodeMap* node_map;
Ddependency_optimizer.h80 std::unique_ptr<NodeMap> node_map_;
Dloop_optimizer.cc76 std::unique_ptr<NodeMap> node_map_;
382 node_map_.reset(new NodeMap(optimized_graph_)); in Optimize()
520 NodeMap node_map(optimized_graph); in RemoveStackOps()
598 const NodeDef& switch_node, const NodeMap& node_map, in CheckForDeadFanout()
749 NodeMap node_map(optimized_graph); in Optimize()
763 const NodeMap& node_map, const absl::flat_hash_set<string>& feed_nodes, in RemoveDeadBranches()
Darithmetic_optimizer_test.cc121 NodeMap node_map(&output); in TEST_F()
161 NodeMap node_map(&output); in TEST_F()
192 NodeMap node_map(&output); in TEST_F()
237 NodeMap node_map(&output); in TEST_F()
386 NodeMap node_map(&output); in TEST_F()
433 NodeMap node_map(&output); in TEST_F()
497 NodeMap node_map(&output); in TEST_F()
584 NodeMap node_map(&output); in TEST_F()
667 NodeMap node_map(&output); in TEST_F()
723 NodeMap node_map(&output); in TEST_F()
[all …]
Dconstant_folding.h45 NodeMap* node_map);
309 std::unique_ptr<NodeMap> node_map_;
Darithmetic_optimizer.h134 std::unique_ptr<NodeMap> node_map_;
/external/tensorflow/tensorflow/core/grappler/utils/
Dtransitive_fanin_test.cc71 NodeMap node_map(&output_graph); in TEST_F()
90 NodeMap node_map(&output_graph); in TEST_F()
111 NodeMap node_map(&output_graph); in TEST_F()
Dcolocation.cc78 NodeMap node_map(graph); in ReassignColocation()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dgraph_compiler_util.cc47 typedef std::unordered_map<string, Node*> NodeMap; typedef
53 Status AddArgNodes(Graph* graph, const NodeMap& node_map, in AddArgNodes()
111 Status AddRetvalNodes(Graph* graph, const NodeMap& node_map, in AddRetvalNodes()
148 NodeMap node_map; in RewriteAndPruneGraph()
/external/swiftshader/third_party/llvm-7.0/llvm/lib/Analysis/
DLazyCallGraph.cpp194 : BPA(std::move(G.BPA)), NodeMap(std::move(G.NodeMap)), in LazyCallGraph()
203 NodeMap = std::move(G.NodeMap); in operator =()
1442 assert(G->NodeMap.find(&NewF) == G->NodeMap.end() && in replaceNodeFunction()
1459 G->NodeMap.erase(&OldF); in replaceNodeFunction()
1460 G->NodeMap[&NewF] = &N; in replaceNodeFunction()
1490 auto NI = NodeMap.find(&F); in removeDeadFunction()
1491 if (NI == NodeMap.end()) in removeDeadFunction()
1496 NodeMap.erase(NI); in removeDeadFunction()
1550 for (auto &FunctionNodePair : NodeMap) in updateGraphPtrs()
/external/libchrome/mojo/core/
Dnode_controller.h136 using NodeMap = variable
261 NodeMap peers_;
320 NodeMap pending_invitations_;
/external/swiftshader/third_party/llvm-7.0/llvm/lib/IR/
DModuleSummaryIndex.cpp232 DenseMap<GlobalValue::GUID, std::vector<uint64_t>> NodeMap; in exportToDot() local
284 NodeMap[SummaryIt.first].push_back(ModId); in exportToDot()
331 auto &ModList = NodeMap[E.Dst]; in exportToDot()
/external/clang/include/clang/ASTMatchers/
DASTMatchersInternal.h127 NodeMap[ID] = DynNode; in addNode()
136 IDToNodeMap::const_iterator It = NodeMap.find(ID); in getNodeAs()
137 if (It == NodeMap.end()) { in getNodeAs()
144 IDToNodeMap::const_iterator It = NodeMap.find(ID); in getNode()
145 if (It == NodeMap.end()) { in getNode()
153 return NodeMap < Other.NodeMap;
164 return NodeMap; in getMap()
170 for (const auto &IDAndNode : NodeMap) { in isComparable()
178 IDToNodeMap NodeMap;
/external/swiftshader/third_party/subzero/src/
DIceConverter.cpp104 NodeMap.clear(); in convertFunction()
188 if (NodeMap.find(BB) == NodeMap.end()) { in mapBasicBlockToNode()
189 NodeMap[BB] = Func->makeNode(); in mapBasicBlockToNode()
191 NodeMap[BB]->setName(BB->getName()); in mapBasicBlockToNode()
193 return NodeMap[BB]; in mapBasicBlockToNode()
649 std::map<const BasicBlock *, Ice::CfgNode *> NodeMap; member in __anon46c03eb20111::LLVM2ICEFunctionConverter
/external/swiftshader/third_party/llvm-7.0/llvm/lib/CodeGen/SelectionDAG/
DSelectionDAGBuilder.h97 DenseMap<const Value*, SDValue> NodeMap; variable
692 if (NodeMap.find(V) == NodeMap.end()) in getNodeForIRValue()
694 return NodeMap[V].getNode(); in getNodeForIRValue()
701 SDValue &N = NodeMap[V]; in setValue()

12