Searched refs:node_name_index (Results 1 – 4 of 4) sorted by relevance
205 auto node_name_index = g->BuildNodeNameIndex(); in PostprocessDataEdgesBetweenOutsideCompilations() local213 auto iter = node_name_index.find(node_name); in PostprocessDataEdgesBetweenOutsideCompilations()214 if (iter == node_name_index.end()) { in PostprocessDataEdgesBetweenOutsideCompilations()258 node_name_index[replace_node->name()] = replace_node; in PostprocessDataEdgesBetweenOutsideCompilations()271 auto node_name_index = g->BuildNodeNameIndex(); in PostprocessControlEdgesBetweenOutsideCompilations() local288 auto iter = node_name_index.find(control_input); in PostprocessControlEdgesBetweenOutsideCompilations()289 if (iter == node_name_index.end()) { in PostprocessControlEdgesBetweenOutsideCompilations()321 auto node_name_index = g->BuildNodeNameIndex(); in PerformStaticShapeInferenceBeforeEncapsulation() local329 Node* n = node_name_index[iter.first]; in PerformStaticShapeInferenceBeforeEncapsulation()
311 auto node_name_index = xla_fbody->graph->BuildNodeNameIndex(); in TEST_F() local314 Node *host_compute_0 = node_name_index["outside_compilation_0_host_compute"]; in TEST_F()316 Node *host_compute_1 = node_name_index["outside_compilation_1_host_compute"]; in TEST_F()513 auto node_name_index = host_graph->BuildNodeNameIndex(); in TEST_F() local514 EXPECT_NE(node_name_index.find("const1"), node_name_index.end()); in TEST_F()616 auto node_name_index = host_graph->BuildNodeNameIndex(); in TEST_F() local619 Node *recv_if_pred_node = node_name_index["recv_oc_if_pred_if"]; in TEST_F()624 Node *if_oc_node = node_name_index["oc_if_if"]; in TEST_F()666 auto node_name_index = xla_graph->BuildNodeNameIndex(); in TEST_F() local670 Node *send_if_pred_node = node_name_index["send_oc_if_pred_if"]; in TEST_F()[all …]
653 auto node_name_index = graph_.BuildNodeNameIndex(); in TEST_F() local654 EXPECT_EQ(node_name_index.size(), 5); in TEST_F()658 EXPECT_NE(node_name_index.find(node_name), node_name_index.end()); in TEST_F()659 EXPECT_EQ(node_name_index[node_name], FindNode(node_name)); in TEST_F()
401 auto node_name_index = graph.BuildNodeNameIndex(); in TEST() local402 Node* while_node = node_name_index["while"]; in TEST()