Searched refs:host_graph (Results 1 – 2 of 2) sorted by relevance
984 FunctionLibraryDefinition* fld, std::unique_ptr<Graph>* host_graph) { in ConstructHostGraph() argument985 host_graph->reset(new Graph(fld)); in ConstructHostGraph()994 Node* sequencer = (*host_graph)->AddNode(sequencer_def, &s); in ConstructHostGraph()1000 AddHostComputeKeyPlaceholder(xla_cluster_name, host_graph->get())); in ConstructHostGraph()1031 node_map[host_fbody->graph->source_node()] = (*host_graph)->source_node(); in ConstructHostGraph()1032 node_map[host_fbody->graph->sink_node()] = (*host_graph)->sink_node(); in ConstructHostGraph()1054 copy = (*host_graph)->AddNode(copy_def, &s); in ConstructHostGraph()1069 (*host_graph) in ConstructHostGraph()1076 (*host_graph)->AddControlEdge(copy, sequencer); in ConstructHostGraph()1086 TF_RETURN_IF_ERROR(ResetDeviceOrdinalToPlaceholderValue(host_graph->get())); in ConstructHostGraph()[all …]
349 Graph *host_graph = host_fbody->graph; in TEST_F() local351 for (Node *n : host_graph->nodes()) { in TEST_F()367 for (Node *n : host_graph->nodes()) { in TEST_F()506 Graph *host_graph = host_fbody->graph; in TEST_F() local507 auto node_name_index = host_graph->BuildNodeNameIndex(); in TEST_F()656 Graph *host_graph = host_fbody->graph; in TEST_F() local657 auto node_name_index = host_graph->BuildNodeNameIndex(); in TEST_F()792 Graph *host_graph = host_fbody->graph; in TEST_F() local793 auto node_name_index = host_graph->BuildNodeNameIndex(); in TEST_F()