Searched refs:node_name_index (Results 1 – 9 of 9) sorted by relevance
211 auto node_name_index = g->BuildNodeNameIndex(); in PostprocessDataEdgesBetweenOutsideCompilations() local219 auto iter = node_name_index.find(node_name); in PostprocessDataEdgesBetweenOutsideCompilations()220 if (iter == node_name_index.end()) { in PostprocessDataEdgesBetweenOutsideCompilations()264 node_name_index[replace_node->name()] = replace_node; in PostprocessDataEdgesBetweenOutsideCompilations()277 auto node_name_index = g->BuildNodeNameIndex(); in PostprocessControlEdgesBetweenOutsideCompilations() local294 auto iter = node_name_index.find(control_input); in PostprocessControlEdgesBetweenOutsideCompilations()295 if (iter == node_name_index.end()) { in PostprocessControlEdgesBetweenOutsideCompilations()331 auto node_name_index = g->BuildNodeNameIndex(); in PerformStaticShapeInferenceBeforeEncapsulation() local339 Node* n = node_name_index[iter.first]; in PerformStaticShapeInferenceBeforeEncapsulation()
307 auto node_name_index = xla_fbody->graph->BuildNodeNameIndex(); in TEST_F() local310 Node *host_compute_0 = node_name_index["outside_compilation_0_host_compute"]; in TEST_F()312 Node *host_compute_1 = node_name_index["outside_compilation_1_host_compute"]; in TEST_F()507 auto node_name_index = host_graph->BuildNodeNameIndex(); in TEST_F() local510 Node *recv_if_pred_node = node_name_index["recv_oc_if_pred_if"]; in TEST_F()515 Node *if_oc_node = node_name_index["oc_if_if"]; in TEST_F()552 auto node_name_index = xla_graph->BuildNodeNameIndex(); in TEST_F() local556 Node *send_if_pred_node = node_name_index["send_oc_if_pred_if"]; in TEST_F()569 Node *if_node = node_name_index["if"]; in TEST_F()657 auto node_name_index = host_graph->BuildNodeNameIndex(); in TEST_F() local[all …]
136 auto node_name_index = g->BuildNodeNameIndex(); in TEST() local137 const Node *if_node = node_name_index.at("if"); in TEST()144 const Node *ret0_node = node_name_index.at("ret0"); in TEST()148 const Node *ret1_node = node_name_index.at("ret1"); in TEST()154 const Node *while_node = node_name_index.at("while"); in TEST()160 const Node *ret2_node = node_name_index.at("ret2"); in TEST()164 const Node *ret3_node = node_name_index.at("ret3"); in TEST()
2510 auto node_name_index = g->BuildNodeNameIndex(); in ExtractOutsideCompilation() local2530 Node* pivot_node = node_name_index[pivot_name]; in ExtractOutsideCompilation()
142 std::unordered_map<std::string, Node*> node_name_index = in UpdateResourceUsageFromFunctionBodyAnalysis() local150 const Node* arg_src = node_name_index[src_node_info.node_name_]; in UpdateResourceUsageFromFunctionBodyAnalysis()165 const Node* ret_user = node_name_index[dst_node_info.node_name_]; in UpdateResourceUsageFromFunctionBodyAnalysis()
406 auto node_name_index = graph.BuildNodeNameIndex(); in TEST() local407 Node* while_node = node_name_index["while"]; in TEST()497 auto node_name_index = bwd_fbody->graph->BuildNodeNameIndex(); in TEST() local498 const Node* identity = node_name_index.at("identity"); in TEST()
645 auto node_name_index = graph->BuildNodeNameIndex(); in GetGraph() local650 const auto& node_iter = node_name_index.find(node_name); in GetGraph()651 if (node_iter != node_name_index.end()) { in GetGraph()668 auto node_name_index = graph->BuildNodeNameIndex(); in GetGraph() local673 const auto& node_iter = node_name_index.find(node_name); in GetGraph()674 if (node_iter != node_name_index.end()) { in GetGraph()
1762 auto node_name_index = graph->BuildNodeNameIndex(); in TEST_F() local1763 Node* ret_node = node_name_index["B"]; in TEST_F()
654 auto node_name_index = graph_.BuildNodeNameIndex(); in TEST_F() local655 EXPECT_EQ(node_name_index.size(), 5); in TEST_F()659 EXPECT_NE(node_name_index.find(node_name), node_name_index.end()); in TEST_F()660 EXPECT_EQ(node_name_index[node_name], FindNode(node_name)); in TEST_F()